-
Notifications
You must be signed in to change notification settings - Fork 74
/
bipartite.py
40 lines (31 loc) · 885 Bytes
/
bipartite.py
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
# Uses python3
import sys
def bipartite(adj):
visited = [False] * len(adj)
visited[0] = True
partition = [-1] * len(adj)
partition[0] = 0
queue = []
queue.append(0)
while queue:
v = queue.pop(0)
for u in adj[v]:
if partition[u] == partition[v]:
return 0
else:
if not visited[u]:
visited[u] = True
partition[u] = 1 - partition[v]
queue.append(u)
return 1
if __name__ == '__main__':
user_input = sys.stdin.read()
data = list(map(int, user_input.split()))
n, m = data[0:2]
data = data[2:]
edges = list(zip(data[0:(2 * m):2], data[1:(2 * m):2]))
adj = [[] for _ in range(n)]
for (a, b) in edges:
adj[a - 1].append(b - 1)
adj[b - 1].append(a - 1)
print(bipartite(adj))