判断二分图
给定一个无向图graph,当这个图为二分图时返回true。
如果我们能将一个图的节点集合分割成两个独立的子集A和B,并使图中的每一条边的两个节点一个来自A集合,一个来自B集合,我们就将这个图称为二分图。
graph将会以邻接表方式给出,graph[i]表示图中与节点i相连的所有节点。每个节点都是一个在0到graph.length-1之间的整数。这图中没有自环和平行边: graph[i] 中不存在i,并且graph[i]中没有重复的值。
解题思路
1、设任意两个节点 u 和 v,如果它们之间有一条边直接相连,那么 u和 v 必须属于不同的集合。
2、如果给定的无向图连通,那么我们就可以任选一个节点开始,给它染成红色。随后我们对整个图进行遍历,将该节点直接相连的所有节点染成绿色,表示这些节点不能与起始节点属于同一个集合。我们再将这些绿色节点直接相连的所有节点染成红色,以此类推,直到无向图中的每个节点均被染色。
示例如图:
3、如果我们能够成功染色,那么红色和绿色的节点各属于一个集合,这个无向图就是一个二分图。
4、如果我们未能成功染色,即在染色的过程中,某一时刻访问到了一个已经染色的节点,并且它的颜色与我们将要给它染上的颜色不相同,也就说明这个无向图不是一个二分图。
算法流程
1、我们任选一个节点开始,将其染成红色,并从该节点开始对整个无向图进行遍历;
2、在遍历的过程中,如果我们通过节点 u遍历到了节点 v(即 u 和 v 在图中有一条边直接相连),那么会有两种情况:
如果 v 未被染色,那么我们将其染成与 u 不同的颜色,并对 v直接相连的节点进行遍历;
如果 v 被染色,并且颜色与 u 相同,那么说明给定的无向图不是二分图。我们可以直接退出遍历并返回 False 作为答案。
3、当遍历结束时,说明给定的无向图是二分图,返回 True 作为答案。
深度优先搜索
class Solution:
def isBipartite(self, graph: List[List[int]]) -> bool: