[JAVA/백준] 1260번 : DFS와 BFS

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

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

<Output>

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

 

<examples>
-입력

4 5 1
1 2
1 3
1 4
2 4
3 4

-출력

1 2 4 3
1 2 3 4

 

[풀이]

import java.util.*;

public class Main {
    static boolean visited[];
    static ArrayList<Integer>[] A;
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int v = sc.nextInt();
        A = new ArrayList[n+1];

        for (int i=1;i<=n;i++){
            A[i] = new ArrayList<>();
        }//하나의 인덱스에 여러개
        for (int i=0;i<m;i++){
            int s = sc.nextInt();
            int e = sc.nextInt();
            A[s].add(e);
            A[e].add(s);//양방향리스트
        }
        for (int i=1;i<=n;i++){
            Collections.sort(A[i]);
        }//인덱스 순회하며 각 방에 담긴 숫자들 오름차순(기본)으로 정렬

        visited = new boolean[n+1];

        DFS(v);
        System.out.println();
        visited = new boolean[n+1];

        BFS(v);
        System.out.println();


    }

    private static void BFS(int v) {
        Queue<Integer> queue = new LinkedList<Integer>();
        queue.add(v);
        visited[v] = true;

        while (!queue.isEmpty()){
            int now = queue.poll();
            System.out.print(now+" ");
            for (int i : A[now]){
                if (!visited[i]){
                    visited[i] = true;
                    queue.add(i);
                }
            }
        }
    }

    private static void DFS(int v) {
        System.out.print(v+" ");
        visited[v] = true;
        for (int i : A[v]){
            if (!visited[i]){
                DFS(i);
            }
        }
    }
}

---

 

기초적인 BFS, DFS 문제.

비슷하지만 순회 방식이 다른 것을 이용하여 풀면 된다.

DFS는 재귀 형태로 구성하였고 방문 노드를 오름차순 순서대로 출력만 해주면 되어서 간단히 해결.

BFS는 큐를 이용해 poll하여 값을 출력해주고

visited에 false 상태라면 true로 바꿔주며

해당 인덱스를 큐에 집어넣어준다. -> 재귀함수로 불러내는 게 아니라 큐가 빈 상태이기 전까지

while 반복문 돌며 poll + 출력해주는 형태.