[백준 1260] DFS와 BFS | 자바

문제

DFS와 BFS

풀이

import java.util.*;

class Main {
  static int n;
  static int m;
  static int v;

  static boolean[] visited;
  static ArrayList<ArrayList<Integer>> graph;

  static void dfs(int x) {
    visited[x] = true;
    System.out.print(x + " ");

    for(int next : graph.get(x)) {
      if(!visited[next]) {
        visited[next] = true;
        dfs(next);
      }
    }
  }

  static void bfs(int start) {
    Queue<Integer> q = new LinkedList<>();
    q.offer(start);
    visited[start] = true;

    while(!q.isEmpty()) {
      int x = q.poll();
      System.out.print(x + " ");

      for(int next : graph.get(x)) {
        if(!visited[next]) {
          q.offer(next);
          visited[next] = true;
        }
      }
    }
  }
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);

    n = sc.nextInt();
    m = sc.nextInt();
    v = sc.nextInt();

    visited = new boolean[n + 1];

    // 그래프 만들기
    graph = new ArrayList<>();
    for(int i = 0; i <= n; i++) {
      graph.add(new ArrayList<Integer>());
    }

    for(int i = 0; i < m; i++) {
      int x = sc.nextInt();
      int y = sc.nextInt();

      graph.get(x).add(y);
      graph.get(y).add(x);
    }

    // 그래프 정렬하기
    for(int i = 0; i < graph.size(); i++) {
      Collections.sort(graph.get(i));
    }

    dfs(v);

    System.out.println("");

    visited = new boolean[n + 1];

    bfs(v);
  }
}

• dfs는 재귀함수, bfs는 큐로 구현합니다.
• 계속 푸니깐 이제 익숙해지는 느낌..