[Java] 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는 큐를 활용하여 문제를 풀이한다.

자세한 흐름은 주석으로 담는다.

자바 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;

public class DfsAndBfs {
    private static int N;
    private static int M;
    private static int V;

    private static boolean[][] lines;
    private static boolean[] visited;

    private static String dfsN;
    private static String bfsN;

    private static StringBuilder sb = new StringBuilder();
    private static Queue<Integer> q = new LinkedList<>();

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] nums = br.readLine().split(" ");

// 노드 수, 간선 수, 탐색 시작 수
        N = Integer.parseInt(nums[0]);
        M = Integer.parseInt(nums[1]);
        V = Integer.parseInt(nums[2]);

// 간선의 연결 배열, 탐색 방문 체크 배열
        lines = new boolean[N + 1][N + 1];
        visited = new boolean[N + 1];

// 간선 입력
        for(int i = 0; i < M; i++) {
            String[] pos = br.readLine().split(" ");
            lines[Integer.parseInt(pos[0])][Integer.parseInt(pos[1])] = true;
            lines[Integer.parseInt(pos[1])][Integer.parseInt(pos[0])] = true;
        }

// dfs 실행, 실행 후 StringBuilder에 입력된 내용 저장
// 방문 처리 초기화, StringBuilder 초기화
        dfs(V);
        dfsN = sb.toString();
        Arrays.fill(visited, false);
        sb.delete(0, sb.length());

// bfs 실행, 결과 저장
        bfs(V);
        bfsN = sb.toString();

// 최종 답안 출력
        System.out.println(dfsN);
        System.out.println(bfsN);
    }

    public static void dfs(int start) {
// 시작 노드(방문 노드) 방문 처리 및 Stringbuilder 저장
        visited[start] = true;
        sb.append(start).append(" ");

// 방문 노드와 연결되고 아직 방문하지 않은 노드를 이용한 재귀
// 문제 2번 예제 기준
// 3
// 3 1
// 3 1 2
// 3 1 2 5
// 3 1 2 5 4
        for(int i = 0 ; i <= N ; i++) {
            if(lines[start][i] && !visited[i])
                dfs(i);
        }

    }

    public static void bfs(int start) {
    // 큐에 시작(방문)노드 삽입 및 방문처리
        q.add(start);
        visited[start] = true;

	// 큐가 완전히 빌때 까지 실행
        while(!q.isEmpty()) {

	// 큐에서 수를 뽑고 sb 저장
            start = q.poll();
            sb.append(start).append(" ");
	// 방문한 적 없고 노드와 연결된 다른 노드를 큐에 삽입 및 방문처리
    // 큐에 새로운 노드가 들어갔기 때문에 해당 노드를 기준으로 다시 방문처리를 한다.
            for(int i = 1 ; i <= N; i++) {
                if(lines[start][i] && !visited[i]) {
                    q.add(i);
                    visited[i] = true;
                }
            }
        }
    }
}