백준 BOJ 2606번 - 바이러스

---

접근 방법

 

DFS와 BFS 둘 다 사용할 수 있는 문제다. DFS, BFS를 활용할 줄 안다면 쉽게 풀 수 있는 문제다.

 

필자는 BFS를 사용해서 풀었다.

 

DFS, BFS를 정리해놓은 글이다.

https://edvedv.tistory.com/4?category=1038364 

백준 BOJ 1012번

 

// Class 3에는 bfs가 굉장히 많은 느낌

---

코드

import java.util.*;

public class Main {
	static int node[][];
	static boolean visit[];
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);	
		int N = in.nextInt();
		int M = in.nextInt();
		
      node = new int[N+1][N+1];
		visit = new boolean[N+1];

		for(int i = 0 ;i < M ;i++) {
			int a = in.nextInt();
			int b = in.nextInt();
			node[a][b] = node[b][a] = 1;
		}
		bfs(1);
	}
	static void bfs(int start) {
		Queue<Integer> q = new LinkedList<>();
		visit[start] = true;
		q.offer(start);
		int count = 0; 
		while(!q.isEmpty()) {
			int temp = q.poll();	
			for(int i = 1; i < node.length; i++) {
				if(node[temp][i] == 1 && !visit[i]) {
					q.offer(i);
					visit[i] = true;
					count++;
				}
			}
		}
		System.out.println(count);
	}
}