1260 - DFS와 BFS[S2]

문제

https://www.acmicpc.net/problem/1260

1260번: DFS와 BFS

문제

그래프를 DFS로 탐색한 결과와 BFS로 탐색한 결과를 출력하는 프로그램을 작성하시오. 단, 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문하고, 더 이상 방문할 수 있는 점이 없는 경우 종료한다. 정점 번호는 1번부터 N번까지이다.

입력

첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사이에 여러 개의 간선이 있을 수 있다. 입력으로 주어지는 간선은 양방향이다.

출력

첫째 줄에 DFS를 수행한 결과를, 그 다음 줄에는 BFS를 수행한 결과를 출력한다. V부터 방문된 점을 순서대로 출력하면 된다.

 

풀이

1. 인접 리스트를 사용하였다.

2. setting의 리스트 생성 후, 자체적으로 sort를 통해 자동으로 dfs에서 최저값이 나오게 하였다.

 

코드

import java.util.*;
import java.io.*;

class Main {
    static int N, M, V;
    static ArrayList<ArrayList<Integer>> list = new ArrayList<>();
    static boolean[] visited;
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
    
    public static void main(String[] args) throws IOException {
        StringTokenizer st = new StringTokenizer(br.readLine());
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());
        V = Integer.parseInt(st.nextToken()) - 1;
        
        visited = new boolean[N];
        setting(); // 그래프 넣기


        dfs(V); // DFS 결과
        Arrays.fill(visited, false); // 방문 기록 초기화
        bw.newLine();
        bfs(); // BFS 결과
        bw.flush();
        bw.close();
    }

    public static void dfs(int v) throws IOException {
        if (visited[v]) return;
        visited[v] = true;
        bw.write((v + 1) + " ");
        for (int i : list.get(v)) {
            if (!visited[i]) dfs(i);
        }
    }

    public static void bfs() throws IOException {
        Queue<Integer> queue = new LinkedList<>();
        queue.offer(V);
        visited[V] = true;

        while (!queue.isEmpty()) {
            int v = queue.poll();
            bw.write((v + 1) + " ");
            for (int i : list.get(v)) {
                if (!visited[i]) {
                    queue.offer(i);
                    visited[i] = true;
                }
            }
        }
    }

    public static void setting() throws IOException {
        for (int i = 0; i < N; i++) { // N개의 노드에 대한 리스트 초기화
            list.add(new ArrayList<>());
        }

        for (int i = 0; i < M; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine()); // 여기서 st 변수를 선언
            int start = Integer.parseInt(st.nextToken()) - 1;
            int end = Integer.parseInt(st.nextToken()) - 1;
            list.get(start).add(end);
            list.get(end).add(start);
        }
        
        for (int i = 0; i < N; i++) {
            Collections.sort(list.get(i)); // 각 리스트 정렬
        }
    }
}