算法题 染色法判断二分图（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