글 읽기 - [java] 트리의 부모 찾기 런타임 에러 도와주세요!

sheepbomb 5년 전 0

간선의 노드가 둘다 없을 경우 stack에 넣어서 처리하는 방법으로 구현했는데

런타임 에러가 발생하네요 ...

도움 부탁드립니다 (_ _)

package com.algoFactory.baekjoonAlgo;

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

public class a_11725 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        int root[] = new int[N + 1];
        root[1] = 1;

        StringTokenizer st;
        Stack<int[]> stack = new Stack<>();
        Stack<int[]> stack2 = new Stack<>();
        for (int i = 0; i < N - 1; i++) {
            st = new StringTokenizer(br.readLine());
            int a = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());

            if (root[a] == 0 && root[b] == 0) {
                int[] arr = {a, b};
                stack.push(arr);
            } else if (root[a] != 0)
                root[b] = a;
            else    //root[b] !=0
                root[a] = b;
        }

        while (!stack.isEmpty() || !stack2.isEmpty()) {
            while (!stack.isEmpty()) {
                int[] arr = stack.pop();
                int a = arr[0];
                int b = arr[1];

                if (root[a] == 0 && root[b] == 0) {
                    arr[0] = a;
                    arr[1] = b;
                    stack2.push(arr);
                } else if (root[a] != 0)
                    root[b] = a;
                else    //root[b] !=0
                    root[a] = b;
            }
            while (!stack2.isEmpty()) {
                int[] arr = stack.pop();
                int a = arr[0];
                int b = arr[1];

                if (root[a] == 0 && root[b] == 0) {
                    arr[0] = a;
                    arr[1] = b;
                    stack.push(arr);
                } else if (root[a] != 0)
                    root[b] = a;
                else    //root[b] !=0
                    root[a] = b;
            }
        }
        for (int i = 2; i <= N; i++)
        {
            System.out.println(root[i]);
        }
    }
}

djm03178 5년 전 1

48번째 줄 stack2가 비어있지 않다는 것만으로 49번째 줄에서 stack을 (2가 아니라) pop하는 것이 안전하다고 보장되나요?

sheepbomb 5년 전 0

@djm03178 오우 ............ ㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠ 진짜 기본적인건데 괜히 질문올렸네요 ㅠㅠㅠㅠㅠ 감사합니다 ...ㅠㅠㅠ

코드 복붙하다 실수한거같습니다 ㅠㅠ

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

[java] 트리의 부모 찾기 런타임 에러 도와주세요!

sheepbomb 5년 전 0

djm03178 5년 전 1

sheepbomb 5년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말