[코드그루] 120. 백준 1325번 : 효율적인 해킹

2025. 1. 14. 21:05·코딩테스트

https://www.acmicpc.net/problem/1325

이차원배열이랑 dfs로 시도해봤는데 시간초과남… 왜 난 리스트를 봐도봐도 기억을 못할까? 노오력이 부족한건가...

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

public class Main {
    static int n, m;
    static ArrayList<ArrayList<Integer>> list;
    static boolean[] visited;
    static int[] cnt;
    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());
        list = new ArrayList<>();
        for (int i=0; i <= n; i++) {
            list.add(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());
            // a -> b일 때 b는 a를 해킹할 수 있다.
            list.get(b).add(a);
        }
        cnt = new int[n + 1];   // 해킹할 수 있는 컴퓨터 수 저장할 배열
        int max = 0;
        for (int i=1; i <= n; i++) {
            bfs(i);
            max = Math.max(max, cnt[i]);
        }
        StringBuilder sb = new StringBuilder();
        for (int i=1; i <= n; i++) {
            if (cnt[i] == max) {    // 해킹 컴퓨터 수 최댓값과 값이 같다면 해당 인덱스 저장
                sb.append(i).append(" ");
            }
        }
        System.out.println(sb);
    }

    public static void bfs(int start) {
        Queue<Integer> queue = new LinkedList<>();
        visited = new boolean[n + 1];
        queue.add(start);
        visited[start] = true;
        while(!queue.isEmpty()) {
            int node = queue.poll();
            // 신뢰관계가 있고 아직 방문하지 않은 경우
            for (int next : list.get(node)) {
                if (!visited[next]) {
                    cnt[start]++;
                    visited[next] = true;
                    queue.add(next);
                }
            }
        }
    }
}

728x90

'코딩테스트' 카테고리의 다른 글

[코드그루] 122. 백준 2343번 : 기타 레슨  (0) 2025.01.17
[코드그루] 121. 백준 3088번 : 화분 부수기  (1) 2025.01.15
[코드그루] 119. 백준 11052번 : 카드 구매하기  (0) 2025.01.13
[코드그루] 118. 백준 9440번 : 숫자 더하기  (0) 2025.01.13
[코드그루] 117. 백준 1080번 : 행렬  (0) 2025.01.13
'코딩테스트' 카테고리의 다른 글
  • [코드그루] 122. 백준 2343번 : 기타 레슨
  • [코드그루] 121. 백준 3088번 : 화분 부수기
  • [코드그루] 119. 백준 11052번 : 카드 구매하기
  • [코드그루] 118. 백준 9440번 : 숫자 더하기
hnajeahi
hnajeahi
개발 스터딩
  • hnajeahi
    hnajeahi-hub
    hnajeahi
  • 전체
    오늘
    어제
    • 전체보기 (96)
      • 개발일기 (3)
      • 코딩테스트 (64)
      • TIL (29)
  • 블로그 메뉴

    • 홈
    • 태그
    • 방명록
  • 링크

  • 공지사항

    • 블로그 이사왔음!
  • 인기 글

  • 태그

    해시
  • 최근 댓글

  • 최근 글

  • hELLO· Designed By정상우.v4.10.3
hnajeahi
[코드그루] 120. 백준 1325번 : 효율적인 해킹
상단으로

티스토리툴바