[백준 1270 / JAVA] DFS와 BFS

https://www.acmicpc.net/problem/1260

1260번: DFS와 BFS

 

기본적인 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;
import java.util.StringTokenizer;

public class Main {
    static int N, M, V;
    static int[][] map;
    static boolean[] check;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String input;
        StringTokenizer st;
        int x, y;
        Queue<Integer> queue = new LinkedList<>();

        input = br.readLine();
        st = new StringTokenizer(input);

        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());
        V = Integer.parseInt(st.nextToken());

        map = new int[N + 1][N + 1];
        check = new boolean[N + 1];

        for (int i = 0; i < M; i++) {
            input = br.readLine();
            st = new StringTokenizer(input);
            x = Integer.parseInt(st.nextToken());
            y = Integer.parseInt(st.nextToken());
            map[x][y] = 1;
            map[y][x] = 1;
        }
        // dfs
        dfs(V);

        // bfs
        // 방문 초기화
        Arrays.fill(check, false);
        queue.add(V);
        check[V] = true;
        System.out.println();
        while (!queue.isEmpty()) {
            V = queue.remove();
            System.out.print(V + " ");
            for (int i = 1; i <= N; i++) {
                if (map[V][i] > 0 && check[i] == false) {
                    check[i] = true;
                    queue.add(i);
                }
            }
        }

    }

    public static void dfs(int v) {
        if (check[v] == true)
            return;
        System.out.print(v + " ");
        check[v] = true;
        for (int i = 1; i <= N; i++) { //연결된 부분 dfs
            if (map[v][i] > 0)
                dfs(i);
        }
    }
}

dfs를 실행한 후 바뀐 check 배열을 다시 false로 바꿔준 후 bfs를 실행한다.