[BOJ S1][#DFS]1325번 효율적인 해킹

효율적인 해킹 성공다국어

문제

해커 김지민은 잘 알려진 어느 회사를 해킹하려고 한다. 이 회사는 N개의 컴퓨터로 이루어져 있다. 김지민은 귀찮기 때문에, 한 번의 해킹으로 여러 개의 컴퓨터를 해킹 할 수 있는 컴퓨터를 해킹하려고 한다.

이 회사의 컴퓨터는 신뢰하는 관계와, 신뢰하지 않는 관계로 이루어져 있는데, A가 B를 신뢰하는 경우에는 B를 해킹하면, A도 해킹할 수 있다는 소리다.

이 회사의 컴퓨터의 신뢰하는 관계가 주어졌을 때, 한 번에 가장 많은 컴퓨터를 해킹할 수 있는 컴퓨터의 번호를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에, N과 M이 들어온다. N은 10,000보다 작거나 같은 자연수, M은 100,000보다 작거나 같은 자연수이다. 둘째 줄부터 M개의 줄에 신뢰하는 관계가 A B와 같은 형식으로 들어오며, "A가 B를 신뢰한다"를 의미한다. 컴퓨터는 1번부터 N번까지 번호가 하나씩 매겨져 있다.

출력

첫째 줄에, 김지민이 한 번에 가장 많은 컴퓨터를 해킹할 수 있는 컴퓨터의 번호를 오름차순으로 출력한다.

예제 입력 1 복사

5 4
3 1
3 2
4 3
5 3

예제 출력 1 복사

1 2

 

 

📝 풀이

신뢰관계 - 단방향 그래프 관계라 생각하고 DFS나 BFS로 풀면 된다.

 

그러나 절대 자바로 풀지마라!!!

자바는 제출마다 시간초과가 랜덤으로 뜬다... 자바에서만 최적화가 필요한 문제이기 때문이다. 

 

 

import java.io.*;
import java.util.*;

public class Main {
    static ArrayList<Integer>[] trustGraph;
    static boolean[] visited;
    static int[] count;
    static int n, m;
    static int result;

    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());
        
        trustGraph = new ArrayList[n + 1];
        visited = new boolean[n + 1];
        count = new int[n + 1];
        
        for (int i = 1; i <= n; i++) {
            trustGraph[i] = new ArrayList<>();
        }

        for (int i = 0; i < m; i++) {
            st = new StringTokenizer(br.readLine());
            int a = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());
            trustGraph[b].add(a); 
        }

        result = 0;

        // 모든 노드에 대해 DFS 수행
        for (int i = 1; i <= n; i++) {
            Arrays.fill(visited, false); 
            int numHacked = dfs(i); // i번 노드에서 시작하는 DFS 수행
            result = Math.max(result, numHacked);
            count[i] = numHacked;
        }

        // 결과 출력
        StringBuilder sb = new StringBuilder();
        for (int i = 1; i <= n; i++) {
            if (count[i] == result) {
                System.out.println(i+" "+ result);
            }
        }
    }

    // DFS 함수
    private static int dfs(int v) {
        visited[v] = true;
        int count = 1; 
        
        for (int next : trustGraph[v]) {
            if (!visited[next]) { // 방문하지 않은 경우에만 DFS 호출
                count += dfs(next);
            }
        }
        
        return count;
    }
}