Algorithm

[BOJ] 1260. DFS와 BFS

프로그래민 2020. 4. 17. 10:50
반응형

그래프를 사용하여 DFS와 BFS를 하는 기본 문제이다.

그래프를 인접리스트로 구현하여 DFS와 BFS각 각 구현하였다. DFS를 구현할 때는 stack을 사용하는 대신 재귀의 형태로 구현하였다. BFS를 구현할 때에는 queue를 사용하여 구현하였다. DFS에서는 방문할때 바로 visit 체크를 해주었고, BFS에서는 queue에 offer하는 동시에 visit 체크를 해주었다.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package online.base;
 
 
public class Main_bj_1260_DFS와BFS {
    
    static int V,E,start;
    static List<Integer>[] graph;
    static boolean[] check;
 
    static Queue<Integer> queue;
    
    public static void main(String[] args) throws Exception {
        BufferedReader br =new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = null;
        
        st=new StringTokenizer(br.readLine());
        V=Integer.parseInt(st.nextToken());
        E=Integer.parseInt(st.nextToken());
        start=Integer.parseInt(st.nextToken());
        
        graph=new ArrayList[V+1];
        for(int i=0;i<=V;i++) {
            graph[i]=new ArrayList<>();
        }
        
        for(int i=0;i<E;i++) {
            st=new StringTokenizer(br.readLine());
            int v1=Integer.parseInt(st.nextToken());
            int v2=Integer.parseInt(st.nextToken());
            
            graph[v1].add(v2);
            graph[v2].add(v1);
        }
        for(int i=0;i<=V;i++) {
            Collections.sort(graph[i]);
        }
        
        check=new boolean[V+1];
        dfs(start);
        System.out.println();
        
        check=new boolean[V+1];
        queue=new LinkedList<>();
        bfs();
        
        
    }
    
    static void dfs(int i) {
        
        check[i]=true;
        System.out.print(i+" ");
        
        for(int next : graph[i]) {
            if(check[next]==false) {
                dfs(next);
            }
        }
    }
    
    static void bfs() {
        queue.offer(start);
        check[start]=true;
        
        while(!queue.isEmpty()) {
            int i = queue.poll();
            System.out.print(i+" ");
            
            for(int next : graph[i]) {
                if(check[next]==false) {
                    queue.offer(next);
                    check[next]=true;
                }
            }
        }
        
    }
}
 
http://colorscripter.com/info#e" target="_blank" style="color:#4f4f4ftext-decoration:none">Colored by Color Scripter
                                                   
반응형

'Algorithm' 카테고리의 다른 글

[BOJ] 1107. 리모컨  (0) 2020.04.18
[BOJ] 13023. ABCDE  (0) 2020.04.17
[BOJ] 14226. 이모티콘  (0) 2020.04.14
[BOJ] 15658. 연산자 끼워넣기(2)  (0) 2020.04.14
[BOJ] 14501. 퇴사  (0) 2020.04.14