[Silver 1] [Java] DFS와 BFS (1260번)

<DFS와 BFS>

문제 설명

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

 

제한 사항

입력

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

출력

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

 

내 생각

DFS와 BFS에 대한 개념을 완벽히 숙지하고 문제를 푸는것을 추천합니다. 개념을 완벽히 숙지해 구현하는데 큰 어려움은 없었습니다. 먼저 DFS는 재귀를 이용해 구현했습니다. 스택으로도 구현이 가능합니다. BFS는 큐를 이용해 구현했습니다. 

 

푼 시간

21분 47초

 

작성 코드

import java.util.*;
 
public class Exam1260 {
    static ArrayList<Integer>[] a;
    static boolean[] c;
    
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        Exam1260 ex = new Exam1260();
        int n = scan.nextInt();
        int m = scan.nextInt();
        int start = scan.nextInt();
        
        a =(ArrayList<Integer>[]) new ArrayList[n+1];    
        for (int i = 1; i <= n; i++) {
            a[i] =new ArrayList<Integer>();            
        }
        
        for (int i = 0; i < m; i++) {                
            int u = scan.nextInt();
            int v = scan.nextInt();
            a[u].add(v);
            a[v].add(u);
        }
        
        for (int i = 1; i <= n; i++) {
            Collections.sort(a[i]);
        }
        
        c = new boolean[n+1];
        dfs(start);
        System.out.println();
        c = new boolean[n+1];
        bfs(start);
    }
    
    public static void dfs(int x) {
        if (c[x])
            return;
        c[x] = true;
        System.out.print(x + " ");
        for (int y : a[x]) {
            if (c[y] == false) {
                dfs(y);
            }
        }
            
        
    }
    
    public static void bfs(int x) {
        Queue<Integer> q = new LinkedList<Integer>();
        q.offer(x);
        c[x] = true;
        while (!q.isEmpty()) {
            int y = q.poll();
            System.out.print(y + " ");
            for (int z : a[y]) {
                if (c[z] == false) {
                    c[z] = true;
                    q.offer(z);
                }
            }
        }
        
    }
}
 

 

실행 결과