python方格染色_判定二部图的着色算法(Python),题,染色法,判断,二分
題目:
給定一個n個點m條邊的無向圖,圖中可能存在重邊和自環。
請你判斷這個圖是否是二分圖。
輸入格式
第一行包含兩個整數n和m。
接下來m行,每行包含兩個整數u和v,表示點u和點v之間存在一條邊。
輸出格式
如果給定圖是二分圖,則輸出“Yes”,否則輸出“No”。
數據范圍
1≤n,m≤10^5
輸入樣例:
4 4
1 3
1 4
2 3
2 4
輸出樣例:
Yes
代碼:
n, m = map(int, input().split())
st = [0]*(n+1) # 染色數組
idx = 0
e, ne = [0]*(2*m+1), [0]*(2*m+1)
h = [-1]*(n+1)
def add(a, b):
global idx
e[idx], h[a], ne[idx] = b, idx, h[a]
idx += 1
def bfs(i, c):
queue = [(i, c)]
hh, tt = 0, 0
st[i] = c
while hh <= tt:
t, c = queue[hh]
hh += 1
i = h[t]
while i != -1:
j = e[i]
if st[j] == 0:
st[j] = 3-c
queue.append((j, 3-c))
tt += 1
elif st[j] == c:
return False
i = ne[i]
return True
for _ in range(m):
a, b = map(int, input().split())
add(a, b)
add(b, a)
flag = True
for i in range(1, n+1):
if st[i] == 0 and not bfs(i, 1):
flag = False; break
if flag: print('Yes')
else: print('No')
總結
以上是生活随笔為你收集整理的python方格染色_判定二部图的着色算法(Python),题,染色法,判断,二分的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: mysql 两个数据库指定表同步_mys
- 下一篇: python面向接口编程_Python