[백준저지] 1260번 : DFS와 BFS

www.acmicpc.net/problem/1260

1. 문제

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

 

입)

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

 

출)

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

 

2. 풀이

import java.util.*;

public class Main {
    // 인접리스트 ArrayList 배열
    static ArrayList<Integer>[] a;
    // 방문여부 배열
    static boolean[] c;
    // DFS
    public static void dfs(int x) {
        // 방문여부 체크
        if (c[x]) {
            return;
        }
        c[x] = true;
        System.out.print(x + " ");
        // 인접리스트 돌면서
        for (int y : a[x]) {
            // 방문 안한 놈은 dfs
            if (c[y] == false) {
                dfs(y);
            }
        }
    }
    // BFS
    public static void bfs(int start) {
        Queue<Integer> q = new LinkedList<Integer>();
        // 큐에 넣고
        q.add(start);
        // 방문했다고 표시
        c[start] = true;
        // 큐가 empty가 될 때까지
        while (!q.isEmpty()) {
            int x = q.poll();
            System.out.print(x + " ");
            // 인접리스트 돌면서
            for (int y : a[x]) {
                // 방문 안했으면
                if (c[y] == false) {
                    // 방문
                    c[y] = true;
                    // 큐에 넣고
                    q.add(y);
                }
            }
        }
    }
    public static void main(String args[]) {
        Scanner sc = new Scanner(System.in);
        // 정점의 개수
        int n = sc.nextInt();
        // 간선의 개수
        int m = sc.nextInt();
        // 시작할 정점의 번호
        int start = sc.nextInt();
        a = (ArrayList<Integer>[]) new ArrayList[n+1];
        for (int i=1; i<=n; i++) {
            // 정점 별로 list를 만든다
            a[i] = new ArrayList<Integer>();
        }
        // 간선을 리스트에 넣는다
        for (int i=0; i<m; i++) {
            int u = sc.nextInt();
            int v = sc.nextInt();
            // 양방향이니까
            a[u].add(v);
            a[v].add(u);
        }
        // 각 리스트를 sorting
        for (int i=1; i<=n; i++) {
            Collections.sort(a[i]);
        }
        c = new boolean[n+1];
        // DFS
        dfs(start);
        System.out.println();
        c = new boolean[n+1];
        // BFS
        bfs(start);
        System.out.println();
    }
}

DFS는 재귀를 이용하고, BFS는 큐를 사용해서 처리한다.

그리고 그래프가 어떤 정점과 간선으로 이루어져있는지 인접리스트를 이용해서 저장하고, 방문 여부를 갖고 있는 배열도 하나 선언.

처음에 ArrayList에 그래프의 모양을 저장하고, 순서대로 DFS와 BFS를 수행.