-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathBicoloring.java
85 lines (75 loc) · 2.7 KB
/
Bicoloring.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
import static java.util.Arrays.stream;
import static java.util.stream.Collectors.toList;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayDeque;
import java.util.Deque;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
public class Bicoloring {
private static final BufferedReader reader = new BufferedReader(
new InputStreamReader(System.in));
private static void add(Map<Integer, Set<Integer>> graph, Integer nodeFrom,
Integer nodeTo) {
if (!graph.containsKey(nodeFrom)) {
graph.put(nodeFrom, new HashSet<Integer>());
}
if (!graph.containsKey(nodeTo)) {
graph.put(nodeTo, new HashSet<Integer>());
}
graph.get(nodeFrom).add(nodeTo);
graph.get(nodeTo).add(nodeFrom);
}
private static boolean solve(
Map<Integer, Set<Integer>> graph) {
if (graph.size() == 0) {
return true;
}
Deque<Integer> next = new ArrayDeque<>();
Set<Integer> visited = new HashSet<>();
Map<Integer, Boolean> colors = new HashMap<>();
boolean color = true;
Integer start = graph.keySet().iterator().next();
next.push(start);
colors.put(start, color);
while (!next.isEmpty()) {
Integer node = next.pop();
color = colors.get(node);
for (Integer adjNode : graph.get(node)) {
if (!visited.contains(adjNode)) {
visited.add(adjNode);
next.addLast(adjNode);
colors.put(adjNode, !color);
} else if (colors.get(adjNode) == color) {
return false;
}
}
}
return true;
}
public static void main(String[] args) throws IOException {
while (true) {
int n = Integer.parseInt(reader.readLine().trim());
if (n == 0) {
break;
}
n = Integer.parseInt(reader.readLine().trim());
Map<Integer, Set<Integer>> graph = new HashMap<>();
for (int j = 0; j < n; ++j) {
List<Integer> toFrom = stream(
reader.readLine().trim().split(" "))
.filter(x -> !x.equals(""))
.map(Integer::parseInt)
.collect(toList());
add(graph, toFrom.get(0), toFrom.get(1));
}
System.out
.println(
solve(graph) ? "BICOLORABLE." : "NOT BICOLORABLE.");
}
}
}