수학과의 좌충우돌 프로그래밍

[C++] BAEKJOON 11724 연결 요소의 개수 본문

알고리즘/C++

[C++] BAEKJOON 11724 연결 요소의 개수

ssung.k 2019. 10. 2. 00:31

https://www.acmicpc.net/problem/11724

 

11724번: 연결 요소의 개수

첫째 줄에 정점의 개수 N과 간선의 개수 M이 주어진다. (1 ≤ N ≤ 1,000, 0 ≤ M ≤ N×(N-1)/2) 둘째 줄부터 M개의 줄에 간선의 양 끝점 u와 v가 주어진다. (1 ≤ u, v ≤ N, u ≠ v) 같은 간선은 한 번만 주어진다.

www.acmicpc.net

dfs 를 이용해서 문제를 해결하였습니다.

arr 배열을 만들어서 i 노드와 j 노드가 연결되었다면 양방향으로 올 수 있으므로 arr[i][j]arr[j][i] 의 값에 1을 넣어주었습니다. 그리고 visited 배열을 만들어서 해당 노드를 순회한 적이 있는지 없는지 유무를 확인해주었습니다.

다른 노드로 건나가기 위해서는 노드를 연결하는 간선이 존재하고, 방문을 하지않은 노드일 경우여야합니다.

#include <iostream>

using namespace std;

int arr[1001][1001];
int visited[1001];

void dfs(int i, int N){
    
    for (int j=1;j<=N;j++){
        if (arr[i][j]&&!visited[j]){
            visited[j] = 1;
            dfs(j,N);
        }
    }
}

int main(int argc, const char * argv[]) {
    int N,M;
    cin >> N >> M;
    
    for (int i=0;i<M;i++){
        int node1, node2;
        cin >> node1 >> node2;
        
        arr[node1][node2] = 1;
        arr[node2][node1] = 1;
    }
    
    int cnt = 0;
    
    for (int i=1;i<=N;i++){
        if (!visited[i]){
            cnt++;
            visited[i] = 1;
            dfs(i,N);
        }
    }
    
    cout << cnt << "\n";
    
    return 0;
}

 

Comments