글 읽기 - BFS 문제풀이 예외 케이스

mh000511 2년 전 0

채점 시 2%에서 오류가 발생하는데, 예외 케이스를 찾지 못하겠습니다. 조언 부탁 드립니다.

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

/**
 * - 삭제 node == root node일 경우 0 출력 후 종료
 * - 삭제 node의 경우, 부모 node와의 간선 추가하지 않음
 */
public class Main {
    public static boolean[] isVisited;
    public static int[][] arr;
    public static int numOfNoP;

    public static void main(String[] args) throws IOException {
        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
        int numOfNode = Integer.parseInt(reader.readLine());
        int rootNode = 0;
        arr = new int[numOfNode][numOfNode];
        StringTokenizer st = new StringTokenizer(reader.readLine(), " ");
        int removeNode = Integer.parseInt(reader.readLine());

        for (int index = 0; index < arr.length; index++) {
            int pNode = Integer.parseInt(st.nextToken());

            // 삭제 node의 경우, 부모 node와의 간선 추가하지 않음
            if (index == removeNode)
                continue;

            if (pNode == -1) {
                rootNode = index;
                continue;
            }
            arr[index][pNode] = 1;
            arr[pNode][index] = 1;
        }

        // 삭제 node == root 노드일 경우 0 출력 후 종료
        if (rootNode == removeNode) {
            System.out.print(0);
            return;
        }

        isVisited = new boolean[arr.length];
        BFS(rootNode);
        System.out.print(numOfNoP);
    }

    public static void BFS(int initNode) {
        Queue<Integer> q = new LinkedList<>();

        isVisited[initNode] = true;
        q.offer(initNode);

        while (!q.isEmpty()) {
            int fromNode = q.poll();
            boolean hasChild = false;

            for (int toNode = 0; toNode < isVisited.length; toNode++) {
                if (!isVisited[toNode] && arr[fromNode][toNode] == 1) {
                    isVisited[toNode] = true;
                    q.offer(toNode);
                    hasChild = true;
                }
            }
            if (!hasChild)
                numOfNoP++;
        }

    }
}

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

문제

문제

출처

ICPC

BFS 문제풀이 예외 케이스

mh000511 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말