BOJ_S3_2606

📁 [S3_2606] 바이러스

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
	static StringTokenizer st;
	static int[][] arr;
	static boolean[] isChecked;
	static int N;
	// 출력할 컴퓨터 수
	static int cnt = 0;

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();

		// 컴퓨터 수
		int N = Integer.parseInt(br.readLine());

		// 컴퓨터 쌍의 수
		int cp = Integer.parseInt(br.readLine());

		// 컴퓨터 쌍 연결 상태 배열 1-> 연결 o , 0-> 연결 x
		arr = new int[N][N];

		// 감염 유무
		isChecked = new boolean[N];

		for (int i = 0; i < cp; i++) {
			st = new StringTokenizer(br.readLine(), " ");
			int first = Integer.parseInt(st.nextToken());
			int second = Integer.parseInt(st.nextToken());

			// 순서 상관 x
			arr[first - 1][second - 1] = 1;
			arr[second - 1][first - 1] = 1;
		}

		// 감염 체크
		dfs(0);

		// 출력
		sb.append(cnt);
		System.out.println(cnt);
	}

	// 감염 체크
	static void dfs(int current) {
		// 기저조건
		if(current == arr.length) {
			return;
		}
		// 처음 시작하는 컴퓨터는 바이러스에 감염되어있음
		isChecked[current] = true;
		
		for (int i = 0; i < arr.length; i++) { 
			// 다음 것이 false ( 감염 안되어 있고 ), i 가 현재 값이 아니고 , 간선이 연결되어 있다면 추가
			if (!isChecked[i] && i != current && arr[current][i] == 1) { 
				cnt++;
				// 계속 간선 따라서 들어가서 파악한다
				dfs(i);
			}
		}
	}
}

🤔 나의 생각

dfs로 풀어주었다.
1번 컴퓨터와 연결되어있는 간선을 끝까지 따라가서 isChecked에 true를 주고 카운팅을 해주었다.
많은 사람들이 자식을 두번 설정해주는 것을 까먹었다는데 여기서는 방향이 없기 때문에 두번 설정해주는 것이 많이 놓칠 수 있는 부분이다.