코딩테스트
백준 11724번 연결 요소의 개수(자바) - DFS
leeeehhjj
2021. 11. 23. 15:13
연결 요소의 개수
시간 제한메모리 제한제출정답맞힌 사람정답 비율
3 초 | 512 MB | 53979 | 25421 | 16603 | 44.000% |
문제
방향 없는 그래프가 주어졌을 때, 연결 요소 (Connected Component)의 개수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 정점의 개수 N과 간선의 개수 M이 주어진다. (1 ≤ N ≤ 1,000, 0 ≤ M ≤ N×(N-1)/2) 둘째 줄부터 M개의 줄에 간선의 양 끝점 u와 v가 주어진다. (1 ≤ u, v ≤ N, u ≠ v) 같은 간선은 한 번만 주어진다.
출력
첫째 줄에 연결 요소의 개수를 출력한다.
예제 입력 1 복사
6 5
1 2
2 5
5 1
3 4
4 6
예제 출력 1 복사
2
예제 입력 2 복사
6 8
1 2
2 5
5 1
3 4
4 6
5 4
2 4
2 3
풀이 :
첫 번째 정점을 시작으로 dfs(bfs도 가능)를 수행하면 첫 번째 정점과 연결된 모든 점들을 방문하며 visited를 true로 체크한다. 그 후 dfs가 끝나면 연결 요소를 하나 찾은 것이므로 연결 요소의 개수를 +1 한다.
첫 번째 정점의 dfs가 끝나고 아직 방문하지 않은 정점이 있다면 그 정점을 기준으로 또 DFS를 수행한다. 그럼 같은 방식으로 그 점과 연결된 모든 점을 방문하게 된다. 이런 방법을 방문하지 않은 정점이 없어질 때까지 수행한 후 최종적으로 연결 요소의 개수를 출력하면 된다.
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
public class Main {
static int n;
static boolean[][] adjacent;
static boolean[] visited;
public static void main(String args[]) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
n = Integer.parseInt(st.nextToken());
int m = Integer.parseInt(st.nextToken());
adjacent = new boolean[n+1][n+1];
visited = new boolean[n+1];
for (int i = 0; i < m; i++) {
st = new StringTokenizer(br.readLine());
int n1 = Integer.parseInt(st.nextToken());
int n2 = Integer.parseInt(st.nextToken());
adjacent[n1][n2] = true;
adjacent[n2][n1] = true;
}
int count = 0;
for (int i = 1; i <= n; i++) {
if (!visited[i]) {
dfs(i);
count++;
}
}
System.out.println(count);
}
static void dfs(int v) {
visited[v] = true;
for (int i = 1; i <= n; i++) {
if (adjacent[v][i] && !visited[i]) {
dfs(i);
}
}
}
}