[BaekJoon] 1260번 - DFS와 BFS [Java][C++]

문제 링크

---

1. 문제 풀이

주어진 그래프를 DFS로 탐색한 결과와 BFS로 탐색한 결과를 출력하는 문제로 정점 번호가 작은 것을 먼저 방문해야 한다. 인접 리스트를 활용했는데 이때 정점 번호가 작은 것을 먼저 방문할 수 있게 트리셋을 활용해서 정렬을 해줬다. 인접 리스트의 경우 양방향 연결을 해야함에 주의해야 한다.

---

2. 코드

1. 풀이 [Java]

출력을 위해 StringBuilder를 활용해줬다.

import java.io.*;
import java.util.*;

public class Main {

    static List<Set<Integer>> adj = new ArrayList<>();
    static StringBuilder sb = new StringBuilder();

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

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

        for (int i = 0; i <= N; i++) {
            adj.add(new TreeSet<>());
        }

        for (int i = 0; i < M; i++) {
            st = new StringTokenizer(br.readLine());
            int x = Integer.parseInt(st.nextToken());
            int y = Integer.parseInt(st.nextToken());
            adj.get(x).add(y);
            adj.get(y).add(x);
        }

        boolean[] visited = new boolean[1 + N];
        dfs(V, visited);
        sb.append("\n");
        bfs(V, N);

        System.out.println(sb);
    }

    static void dfs(int node, boolean[] visited) {
        visited[node] = true;
        sb.append(node).append(" ");

        for (int next : adj.get(node)) {
            if (visited[next]) continue;
            dfs(next, visited);
        }
    }

    static void bfs(int start, int N) {
        Queue<Integer> q = new ArrayDeque<>();
        q.offer(start);

        boolean[] visited = new boolean[1 + N];
        visited[start] = true;

        while (!q.isEmpty()) {
            int node = q.poll();
            sb.append(node).append(" ");

            for (int next : adj.get(node)) {
                if (visited[next]) continue;

                q.offer(next);
                visited[next] = true;
            }
        }
    }
}

2. 풀이 [C++]

#include <bits/stdc++.h>
using namespace std;

vector<set<int>> adj(1 + 1000);
bool visited[1 + 1000];

void dfs(int node) {
    visited[node] = true;
    cout << node << ' ';

    for (int next : adj[node]) {
        if (visited[next]) continue;
        dfs(next);
    }
}

void bfs(int start) {
    queue<int> q;
    q.push(start);

    visited[start] = true;

    while (!q.empty()) {
        int node = q.front();
        q.pop();

        cout << node << ' ';

        for (int next : adj[node]) {
            if (visited[next]) continue;

            q.push(next);
            visited[next] = true;
        }
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, m, v;
    cin >> n >> m >> v;

    for (int i = 0; i < m; i++) {
        int x, y;
        cin >> x >> y;
        adj[x].insert(y);
        adj[y].insert(x);
    }

    dfs(v);
    cout << '\n';

    memset(visited, 0, sizeof(visited));
    bfs(v);
}

---