LeetCode 785. 判断二分图（染色法）

示例 1:
输入: [[1,3], [0,2], [1,3], [0,2]]
输出: true
解释: 
无向图如下:
0----1
|    |
|    |
3----2
我们可以将节点分成两组: {0, 2} 和 {1, 3}。

示例 2:
输入: [[1,2,3], [0,2], [0,1,3], [0,2]]
输出: false
解释: 
无向图如下:
0----1
| \  |
|  \ |
3----2
我们不能将节点分割成两个独立的子集。

注意:
graph 的长度范围为 [1, 100]。
graph[i] 中的元素的范围为 [0, graph.length - 1]。
graph[i] 不会包含 i 或者有重复的值。
图是无向的: 如果j 在 graph[i]里边, 那么 i 也会在 graph[j]里边。

class Solution {
public:
    bool isBipartite(vector<vector<int>>& graph) {
    	int n = graph.size();
    	vector<int> color(n, 0);
        for(int i = 0; i < n; ++i)
    	{
    		if(color[i] != 0)//染过了，跳过
    			continue;
    		queue<int> q;
	    	q.push(i);
	        color[i] = 1;//染成颜色1，开始bfs
	    	int curColor = 1, newColor, id, k;
	    	while(!q.empty())
	    	{
    			id = q.front();
    			q.pop();
    			curColor = color[id];
                newColor = curColor==1 ? 2 : 1;
    			for(k = 0; k < graph[id].size(); ++k)
    			{
    				if(color[graph[id][k]] == 0)
    				{
    					q.push(graph[id][k]);
                        color[graph[id][k]] = newColor;
    				}
    				else if(color[graph[id][k]] == curColor)
    					return false;
    			}
    		}
    	}
    	return true;
    }
};