-
Notifications
You must be signed in to change notification settings - Fork 0
/
BFS_DFS.java
125 lines (89 loc) · 3.05 KB
/
BFS_DFS.java
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;
import java.util.Queue;
import java.util.Stack;
class Node_1260 {
private int number;
private List<Integer> link;
public Node_1260(int number) {
this.number = number;
this.link = new LinkedList<>();
}
public int getNumber() {
return this.number;
}
public List<Integer> getLink() {
return this.link;
}
}
class Compare implements Comparator<Integer> {
@Override
public int compare(Integer o1, Integer o2) {
return o2 - o1;
}
}
class CompareSort implements Comparator<Integer> {
@Override
public int compare(Integer o1, Integer o2) {
return o1 - o2;
}
}
public class BFS_DFS {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine(), " ");
int nodeNumber = Integer.parseInt(st.nextToken());
int lineNumber = Integer.parseInt(st.nextToken());
int startNumber = Integer.parseInt(st.nextToken());
Node_1260[] node = new Node_1260[nodeNumber+1];
boolean[] visit = new boolean[nodeNumber + 1];
for(int i = 1; i <= nodeNumber; i++) {
node[i] = new Node_1260(i);
}
for(int i = 0; i < lineNumber; i++) {
st = new StringTokenizer(br.readLine(), " ");
int num1 = Integer.parseInt(st.nextToken());
int num2 = Integer.parseInt(st.nextToken());
node[num1].getLink().add(num2);
node[num2].getLink().add(num1);
}
getDFS(node, visit, startNumber);
visit = new boolean[nodeNumber + 1];
System.out.println();
getBFS(node, visit, startNumber);
}
public static void getDFS(Node_1260[] node, boolean[] visit, int start) {
Stack<Integer> stack = new Stack<>();
stack.push(start);
while(!stack.isEmpty()) {
Node_1260 value = node[stack.pop()];
if(visit[value.getNumber()]) {
continue;
}
visit[value.getNumber()] = true;
List<Integer> lst = value.getLink();
Collections.sort(lst, new Compare());
for(int s : lst) {
stack.push(s);
}
System.out.print(value.getNumber() + " ");
}
}
public static void getBFS(Node_1260[] node, boolean[] visit, int start) {
Queue<Integer> queue = new LinkedList<>();
queue.offer(start);
while(!queue.isEmpty()) {
Node_1260 value = node[queue.poll()];
if(visit[value.getNumber()]) continue;
visit[value.getNumber()] = true;
List<Integer> lst = value.getLink();
Collections.sort(lst, new CompareSort());
for(int s : value.getLink()) {
queue.offer(s);
}
System.out.print(value.getNumber() + " ");
}
}
}