백준 11724: 연결 요소의 개수 [Java] - 포포
2022. 5. 19. 19:07ㆍ알고리즘/BFS 알고리즘
문제
방향 없는 그래프가 주어졌을 때, 연결 요소 (Connected Component)의 개수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 정점의 개수 N과 간선의 개수 M이 주어진다. (1 ≤ N ≤ 1,000, 0 ≤ M ≤ N×(N-1)/2) 둘째 줄부터 M개의 줄에 간선의 양 끝점 u와 v가 주어진다. (1 ≤ u, v ≤ N, u ≠ v) 같은 간선은 한 번만 주어진다.
출력
첫째 줄에 연결 요소의 개수를 출력한다.
내 제출
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;
public class No11724 {
static int n,m;
static int[][] arr;
static boolean[] visit;
static int count;
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(" "));
m = Integer.parseInt(st.nextToken(" "));
arr = new int[n+1][n+1];
visit = new boolean[n+1];
count = 0;
for (int i = 0; i < m; i++) {
st = new StringTokenizer(br.readLine());
int a = Integer.parseInt(st.nextToken(" "));
int b = Integer.parseInt(st.nextToken(" "));
arr[a][b] = arr[b][a] = 1;
}
for(int i = 1; i<n+1; i++){
if(!visit[i]){
bfs(i);
count++;
}
}
System.out.println(count);
}
private static void dfs(int i) {
visit[i] = true;
for (int j = 1; j < n + 1; j++) {
if(arr[i][j] == 1 && !visit[j]){
dfs(j);
}
}
}
private static void bfs(int i){
Queue<Integer> qu = new LinkedList<>();
qu.offer(i);
visit[i] = true;
while (!qu.isEmpty()) {
int p = qu.poll();
for(int j = 1; j<= n; j++){
if(arr[p][j] == 1 && !visit[j]){
qu.offer(j);
visit[j] = true;
}
}
}
}
}
bfs와 dfs로 둘 다 구현하였다.
방문하지 않은 정점이 있다면 dfs, bfs 탐색을 시작하고, count를 늘려준다.
bfs, dfs 메서드에서는 다른 정점을 방문하면, 추후에 재탐색하는 경우가 없도록 visit여부를 true로 지정해야 한다.
'알고리즘 > BFS 알고리즘' 카테고리의 다른 글
백준 14502: 연구소 [Java] - 포포 (0) | 2022.10.09 |
---|---|
백준 11725: 트리의 부모 찾기 [Java] -포포 (0) | 2022.09.13 |
백준 12716: 돌다리 [Java] - 포포 (0) | 2022.05.17 |
백준 1388: 바닥 장식 [Java] - 포포 (0) | 2022.05.16 |
백준 14716: 현수막 [Java] - 포포 (0) | 2022.05.07 |