백준 2606번(바이러스)[JAVA]

풀이

1번 컴퓨터가 바이러스 걸렸을 때, 연결되어있어서 같이 바이러스에 걸리게 되는 컴퓨터의 수는, dfs(1)을 하였을 때 방문되는 노드의 개수와 같다. 즉, visited 배열에서 true의 개수를 세어주면 된다.
결과에서 result-1을 해주는 이유는 1이 포함되어 있기 때문이다.

소스 코드

import java.util.*;

public class 바이러스 {
	static boolean visited[];
	static ArrayList<Integer>[] arr;
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int m = sc.nextInt();
		
		arr = new ArrayList[n+1];
		visited = new boolean[n+1];
		
		for (int i = 1; i < n+1; i++) {
			arr[i] = new ArrayList<Integer>();
		}
		
		for (int i = 0; i < m; i++) {
			int a = sc.nextInt();
			int b = sc.nextInt();
			
			arr[a].add(b);
			arr[b].add(a);
		}
		
		dfs(1);
		int result = 0;
		for (int i = 0; i < n+1; i++) {
			if(visited[i])
				result++;
		}
		System.out.println(result-1);
	}
	static void dfs(int v) {
		if (visited[v])
			return;
		
		visited[v] = true;
		for (int i : arr[v]) {
			dfs(i);
		}
	}
}

태클 감사합니다.
조언 환영입니다.