백준 DFS, BFS 알고리즘 - DFS와 BFS 구현 1260 (JAVA)

문제

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

입력

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

출력

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

 

 

package searching;
 
import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;
 
public class BfsDfs_1260 {
 
    //인접 리스트 배열 a
    static ArrayList<Integer>[] a;
    //참, 거짓 구분 값 c
    static boolean[] c;
 
    //dfs 구현 메소드
    //입력값 x는 시작하는 정점의 수이다.
    public static void dfs(int x) {
        if (c[x]) {
            return;
        }
 
        //c[x]노드를 방문한 것으로 간주
        c[x] = true;
        System.out.print(x + " ");
        //a[x]의 인접한 모든 수들을 가져온다.
        for (int y : a[x]) {
            if (c[y] == false) {
                dfs(y);
            }
        }
    }
 
    //dfs 구현 메소드
    //입력값 start는 시작하는 정점의 수이다.
    public static void bfs(int start) {
        
        //queue 의 리스트 q를 만든다
        Queue<Integer> q = new LinkedList<Integer>();
 
        //q에 시작 정점을 넣는다.
        q.add(start);
 
        //c[start]를 true로 만든다.
        c[start] = true;
        //q가 비어있지 않으면 맨 앞에 있는 수인 x를 제거한다. 그리고 x를 출력한다. (큐는 선입선출이기 때문에)
        while (!q.isEmpty()) {
            int x = q.remove();
            System.out.print(x + " ");
            //출력된 x가 갖고있는 값 a[x]의 값들을 찾는다.
            //여기서 만약에 c[y]가 false이면 true로 바꾼 다음에 그 값을 큐에 더한다.
            for (int y : a[x]) {
                if (c[y] == false) {
 
                    c[y] = true;
 
                    q.add(y);
                }
            }
        }
    }
 
    @SuppressWarnings("unchecked")
    public static void main(String args[]) {
        
        Scanner sc = new Scanner(System.in);
        
        //1. 정점의 갯수를 입력 받는다.
        int n = sc.nextInt();
        //2. 간석의 갯수를 입력 받는다.
        int m = sc.nextInt();
        //3. 탐색을 시작할 정점의 번호를 입력 받는다.
        int start = sc.nextInt();
 
        //정점의 갯수만큼 배열을 만든 후 정점을 저장하는 리스트를 만든다.
        a = (ArrayList<Integer>[]) new ArrayList[n + 1];
        //for문을 돌려서 정점의 배열을 만든다.
        for (int i = 1; i <= n; i++) {
 
            a[i] = new ArrayList<Integer>();
        }
        
        //두번째 줄 부터 간선의 연결상태를 입력받는다.
        //간선의 갯수(m)만큼 입력 받는다.(만약 간선이 5개라면 5줄을 입력 받는다.)
        //정점이 연결되어 있는 것들을 표기한다. 만약 1 4 입력되어 있다면 1번 정점과 4번 정점이 연결되어 있는다.
        //u를 입력 받으면 a[u]에 v를 더한다.
        //a[v]를 입력 받으면 a[v]에 u를 더한다.
        //인접 리스트를 통해 계산하는 것으로 각각의 배열에 연결 된 것이 연결된 것들을 나타낸다.
        for (int i = 0; i < m; i++) {
            int u = sc.nextInt();
            int v = sc.nextInt();
 
            a[u].add(v);
 
            a[v].add(u);
        }
        //a의 배열을 오름차순으로 정렬한다.
        for (int i = 1; i <= n; i++) {
            Collections.sort(a[i]);
        }
 
        //c는 새로운 참, 거짓 값 배열
        c = new boolean[n + 1];
        //dfs 메소드 실행
        dfs(start);
        System.out.println();
 
        c = new boolean[n + 1];
        //bfs 메소드 실행
        bfs(start);
        System.out.println();
 
    }
 
}