백준 게임 개발 - 1516 [JAVA] - 23년 1월 13일

Denia·2023년 1월 13일
0

코딩테스트 준비

목록 보기
136/201
post-custom-banner

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Deque;
import java.util.List;

public class Main {
    static public void main(String[] args) throws IOException {
        BjSolution sol = new BjSolution();

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int testCase = Integer.parseInt(br.readLine());
        List<String[]> quizArr = new ArrayList<>();
        for (int i = 0; i < testCase; i++) {
            String[] arr = br.readLine().split(" ");
            quizArr.add(arr);
        }

        sol.solution(testCase, quizArr);
    }
}

class BjSolution {
    public void solution(int testCase, List<String[]> quizArr) {
        int[] answer = new int[testCase + 1];
        int[] time = new int[testCase + 1];
        int[] indgree = new int[testCase + 1];
        List<List<Integer>> nextBuildList = new ArrayList<>();
        for (int i = 0; i < testCase + 1; i++) {
            nextBuildList.add(new ArrayList<>());
        }

        for (int i = 0; i < quizArr.size(); i++) {
            String[] arr = quizArr.get(i);
            int curBuildingNum = i + 1;
            time[curBuildingNum] = Integer.parseInt(arr[0]);
            for (int j = 1; j < arr.length; j++) {
                int preBuildingNum = Integer.parseInt(arr[j]);
                if (preBuildingNum == -1) {
                    break;
                }
                nextBuildList.get(preBuildingNum).add(curBuildingNum);
                indgree[curBuildingNum]++;
            }
        }

        Deque<Integer> queue = new ArrayDeque<>();

        for (int i = 1; i < testCase + 1; i++) {
            if (indgree[i] == 0) {
                queue.addLast(i);
                answer[i] = time[i];
            }
        }

        while (!queue.isEmpty()) {
            int curBuilding = queue.removeFirst();

            for (int nextBuilding : nextBuildList.get(curBuilding)) {
                indgree[nextBuilding]--;

                answer[nextBuilding] = Math.max(answer[nextBuilding], (time[curBuilding] + time[nextBuilding]));

                if (indgree[nextBuilding] == 0) {
                    time[nextBuilding] = answer[nextBuilding];
                    queue.addLast(nextBuilding);
                }
            }
        }

        for (int i = 1; i < answer.length; i++) {
            System.out.println(answer[i]);
        }
    }
}

profile
HW -> FW -> Web
post-custom-banner

0개의 댓글