글 읽기 - 어디서 틀리는지 못찾겠습니다.

jjjhhh14 2년 전 0

제출 시 계속 틀리는데, 틀리는 위치를 못찾겠습니다.

public class Main{
    public static void main(String[] args) throws IOException {
        BufferedReader br =
                new BufferedReader(new InputStreamReader(System.in));

        int[] NMV = getIntArr(br.readLine());
        Graph dfs = new Graph();
        Graph bfs = new Graph();

        for (int i=0; i<NMV[1]; i++){
            int[] nodes = getIntArr(br.readLine());
            dfs.addEdge(nodes[0], nodes[1]);
            bfs.addEdge(nodes[0], nodes[1]);

        }
        dfs.dfs(dfs.nodes.get(NMV[2]));
        System.out.println();
        bfs.bfs(bfs.nodes.get(NMV[2]));

    }

    private static int[] getIntArr(String input) {
        return Arrays.stream(input.split(" "))
                .mapToInt(Integer::parseInt).toArray();
    }

    static class Node implements Comparable<Node>{
        int data;
        List<Node> adjacent = new LinkedList<>();
        boolean visited;
        Node(int data){
            this.data = data;
        }

        @Override
        public int compareTo(Node o) {
            return this.data-o.data;
        }
    }


    static class Graph{
        HashMap<Integer, Node> nodes = new HashMap<>();

        public void addEdge(int a1, int a2){
            Node n1 = addNode(a1);
            Node n2 = addNode(a2);
            if (!n1.adjacent.contains(n2)){
                n1.adjacent.add(n2);
            }
            if (!n2.adjacent.contains(n1)){
                n2.adjacent.add(n1);
            }
        }

        private Node addNode(int data) {
            if (nodes.get(data)==null){
                Node newNode = new Node(data);
                nodes.put(data, newNode);
            }
            return nodes.get(data);
        }

        public void dfs(Node node){
            if (node==null || node.visited){
                return;
            }
            node.visited = true;
            System.out.print(node.data + " ");
            Collections.sort(node.adjacent);
            for (Node adjacent: node.adjacent){
                if (!adjacent.visited){
                    dfs(adjacent);
                }
            }
        }

        public void bfs(Node node){
            Queue<Node> queue = new LinkedList<>();
            queue.add(node);
            node.visited = true;
            while (!queue.isEmpty()){
                Node n = queue.poll();
                Collections.sort(node.adjacent);
                for (Node adjacent: n.adjacent){
                    if (!adjacent.visited){
                        adjacent.visited = true;
                        queue.add(adjacent);
                    }
                }
                System.out.print(n.data + " ");
            }

        }

    }
}

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

문제

문제

출처

ICPC

어디서 틀리는지 못찾겠습니다.

jjjhhh14 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말