[ 백준 - Java ] 바이러스 ( 자바 )

( dfs / 바이러스 / 2606 )

[문제]

문제 설명

신종 바이러스인 웜 바이러스는 네트워크를 통해 전파된다. 한 컴퓨터가 웜 바이러스에 걸리면 그 컴퓨터와 네트워크 상에서 연결되어 있는 모든 컴퓨터는 웜 바이러스에 걸리게 된다.

예를 들어 7대의 컴퓨터가 <그림 1>과 같이 네트워크 상에서 연결되어 있다고 하자. 1번 컴퓨터가 웜 바이러스에 걸리면 웜 바이러스는 2번과 5번 컴퓨터를 거쳐 3번과 6번 컴퓨터까지 전파되어 2, 3, 5, 6 네 대의 컴퓨터는 웜 바이러스에 걸리게 된다. 하지만 4번과 7번 컴퓨터는 1번 컴퓨터와 네트워크상에서 연결되어 있지 않기 때문에 영향을 받지 않는다.

어느 날 1번 컴퓨터가 웜 바이러스에 걸렸다. 컴퓨터의 수와 네트워크 상에서 서로 연결되어 있는 정보가 주어질 때, 1번 컴퓨터를 통해 웜 바이러스에 걸리게 되는 컴퓨터의 수를 출력하는 프로그램을 작성하시오.

제한사항

메모리 제한 : 128 MB

입출력 예시

첫째 줄에는 컴퓨터의 수가 주어진다. 컴퓨터의 수는 100 이하이고 각 컴퓨터에는 1번 부터 차례대로 번호가 매겨진다. 둘째 줄에는 네트워크 상에서 직접 연결되어 있는 컴퓨터 쌍의 수가 주어진다. 이어서 그 수만큼 한 줄에 한 쌍씩 네트워크 상에서 직접 연결되어 있는 컴퓨터의 번호 쌍이 주어진다.

 

1번 컴퓨터가 웜 바이러스에 걸렸을 때, 1번 컴퓨터를 통해 웜 바이러스에 걸리게 되는 컴퓨터의 수를 첫째 줄에 출력한다.

---

[코드]

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class Virus2606 {
    static List<Integer>[] arr; // 인접 리스트
    static boolean[] visited;   // 노드 방문여부
    static int answer;          // 결과

    public static void main(String[] args) {

        // === 입력 ===
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();       // 노드 개수
        int len = sc.nextInt();     // 입력받을 개수

        arr = new ArrayList[n + 1];     // 인접리스트 n + 1만큼 생성
        visited = new boolean[n + 1];   // 방문 배열 n + 1만큼 생성
        // 인접 리스트 초기화
        for(int i = 0; i <= n; i++) arr[i] = new ArrayList<>();
        // 인접 리스트 양방향 추가
        for(int i = 0; i < len; i++) {
            int n1 = sc.nextInt();
            int n2 = sc.nextInt();
            arr[n1].add(n2);
            arr[n2].add(n1);
        }
        sc.close();

        // === 구현 ===
        // 여기서는 단순 카운트만 하면 되기때문에 인접리스트를 정렬 할 필요가 없음
        // 감염 시작인 1번 노드는 제외하기 위해 -1 로 초기화
        answer = -1;
        dfs(1);     // 깊이우선 탐색

        // === 결과 ===
        System.out.println(answer);
    }
    // 깊이우선 탐색
    public static void dfs(int idx){
        if(visited[idx]) return;    // 이미 방문했으면 return

        answer++;               // 현재 노드 카운트
        visited[idx] = true;    // 현재 노드 방문처리

        for(int i : arr[idx]) 
        	if(!visited[i]) dfs(i);   // 현재 노드에서 갈 수 있는 노드 깊이우선으로 방문
    }
}

문제 링크

2606번: 바이러스

제가 잘못 알고 있거나 잘못된 부분이 있을 경우 알려주시고 추가로 궁금한 점 있으신 분들도 댓글이나 메일 주시면 성실히 답변해 드리겠습니다.🧑🏻‍💻

감사합니다~😄