Algorithm/baekjoon

DFS와 BFS

마닐라 2022. 2. 18. 14:36

📍 문제 설명

 

💡 접근

기초 DFS/BFS 문제

1.DFS

방문할 수 있는 지점이 여러 개여도 하나만 출력하면 되므로 출력 배열을 따로 만들어 사용안해도 된다.

 

2.BFS

방문 체크만해서 큐에 넣고 빼면된다. 작은 정점부터 방문해야 하므로 정렬 처리가 필요하다.

👩‍💻 코드

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.*;

public class Main {
    static int n, m, v, answer;
    static int[][] board;
    static int[] arr;
    static long[] dp;
    static boolean[] visited;
    static int[] dx = {-1,1,0,0};
    static int[] dy = {0,0,-1,1};
    static ArrayList<ArrayList<Integer>> list = new ArrayList<>();
    static StringBuilder sb = new StringBuilder();

    public static void main(String[] args) {
        Scanner kb = new Scanner(System.in);
        Main T = new Main();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

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

        for(int i = 0; i <= n; i++) {
            list.add(new ArrayList<>());
        }

        for(int i = 0; i < m; i++) {
            int a = kb.nextInt();
            int b = kb.nextInt();
            list.get(a).add(b);
            list.get(b).add(a);
        }
        for(int i = 0; i <= n; i++) Collections.sort(list.get(i));
        visited = new boolean[n+1];
        T.dfs(v);
        sb.append("\n");
        visited = new boolean[n+1];
        T.bfs();
        System.out.println(sb);

    }

    private void dfs(int start) {
        visited[start] = true;
        sb.append(start+ " ");

        for(int i = 0; i < list.get(start).size(); i++) {
            int nv = list.get(start).get(i);
            if(!visited[nv]) dfs(nv);
        }
    }

    private void bfs() {
        Queue<Integer> q = new LinkedList<>();
        q.add(v);
        visited[v] = true;
        while (!q.isEmpty()) {
            int cv = q.poll();
            sb.append(cv + " ");
            for(int i = 0; i < list.get(cv).size(); i++) {
                int nv = list.get(cv).get(i);
                if(!visited[nv]) {
                    q.offer(nv);
                    visited[nv] = true;
                }
            }
        }
    }

    private void solution(int L, int start) {
    }

}

'Algorithm > baekjoon' 카테고리의 다른 글

이분 그래프  (0) 2022.02.18
연결 요소의 개수  (0) 2022.02.18
ABCDE  (0) 2022.02.17
연속합 2  (0) 2022.02.17
가장 긴 바이토닉 부분 수열  (0) 2022.02.16