【leetcode】785. Is Graph Bipartite?

最新推荐文章于 2022-08-18 08:12:06 发布

原创最新推荐文章于 2022-08-18 08:12:06 发布 · 254 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

257 篇文章

订阅专栏

本文介绍了一种用于判断图是否为二分图的算法。通过创建两个集合分别保存图中的两组节点，从任意节点开始，将其放入集合1，然后将其邻接节点放入集合2，以此类推。如果在放置过程中发现某个节点的邻接节点已经存在于同一集合中，则说明该图不是二分图，算法将返回false；否则，当所有节点都被正确放置后，返回true。此算法适用于无向图的二分图判断。

在这里插入图片描述
创建两个集合用于保存两组节点
首先0放在集合1，0的邻节点放在集合2
1的邻节点是0和2，0已经在集合1里面了，把2也放到集合1里面去
3的邻节点是0和2，都在集合1里面。
所有节点都放在了集合1和2中，返回true（如果存放过程中出现矛盾返回false）

提交代码

class Solution {
    public boolean isBipartite(int[][] graph) {
    	Set<Integer> g1=new HashSet<>();
    	Set<Integer> g2=new HashSet<>();
    	Queue<Integer> node=new LinkedList<>();
    	
    	while(g1.size()+g2.size()<graph.length) {
    		for(int i=0;i<graph.length;i++) {
    			if(!g1.contains(i)&&!g2.contains(i)) {
    				g1.add(i);
    				node.add(i);
    				break;
    			}
    		}
	    	while(node.size()>0) {
	    		int curNode=node.poll();
	    		if(g1.contains(curNode)) {
	    			for(int i=0;i<graph[curNode].length;i++) {
	    				if(g1.contains(graph[curNode][i]))
	    					return false;
	    				if(!g2.contains(graph[curNode][i])) {
	    					node.offer(graph[curNode][i]);
	    					g2.add(graph[curNode][i]);
	    				}
	    			}
	    		}else {
	    			for(int i=0;i<graph[curNode].length;i++) {
	    				if(g2.contains(graph[curNode][i]))
	    					return false;
	    				if(!g1.contains(graph[curNode][i])) {
	    					node.offer(graph[curNode][i]);
	    					g1.add(graph[curNode][i]);
	    				}
	    			}
	    		}
	    	}
    	}
    	return true;
    }
}