인프런 커뮤니티 질문&답변

안녕하세요 선생님
숙제로 내주신 문제를 풀어보았습니다.

좀 무식한 방법으로 한 것 같은데 이것 말고는 다른 방법이 떠오르지가 않더라구요.

피드백 부탁드립니다!

public class Main {

    static int[] visited;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int nodeCount = Integer.parseInt(st.nextToken()); // 노드의 수
        int lineCount = Integer.parseInt(st.nextToken()); // 간선의 수

        List<Integer>[] list = new ArrayList[nodeCount + 1];
        for (int i = 1; i <= nodeCount; i++) {
            list[i] = new ArrayList<>();
        }
        visited = new int[nodeCount + 1];

        for (int i = 0; i < lineCount; i++) {
            st = new StringTokenizer(br.readLine());
            int nodeA = Integer.parseInt(st.nextToken());
            int nodeB = Integer.parseInt(st.nextToken());

            list[nodeA].add(nodeB);
        }

        bfs(list, 2);
    }

    private static void bfs(List<Integer>[] list, int target) {
        Queue<List<Integer>> q = new LinkedList<>();
        q.offer(list[1]);
        visited[1] = 1;

        int level = 1;
        while (!q.isEmpty()) {
            boolean isFinished = false;
            int size = q.size();

            loopOut:
            for (int i = 0; i < size; i++) {
                List<Integer> currentNodes = q.poll();
                for (Integer nextNode : currentNodes) {
                    if (nextNode == target) {
                        System.out.println(target + " : " + level);
                        isFinished = true;
                        visited[nextNode] = 0;
                        break loopOut;
                    } else {
                        List<Integer> e = list[nextNode];
                        if (!e.isEmpty() && visited[nextNode] == 0) {
                            visited[nextNode] = 1;
                            q.offer(e);
                        }
                    }
                }
            }

            if (!isFinished) {
                level++;
            } else {
                target++;
                level = 1;
                q.clear();
                q.offer(list[1]);
                for (int i = 2; i < visited.length; i++) {
                    visited[i] = 0;
                }
                visited[1] = 1;
            }
        }
    }
}