1596 · 可能的二分法

给定一组 N 人（编号为 1, 2, ..., N）， 我们想把每个人分进任意大小的两组。

每个人都可能不喜欢其他人，那么他们不应该属于同一组。

形式上，如果 dislikes[i] = [a, b]，表示不允许将编号为 a 和 b 的人归入同一组。

当可以用这种方法将每个人分进两组时，返回 true；否则返回 false。

1 <= N <= 2000
0 <= dislikes.length <= 10000
1 <= dislikes[i][j] <= N
dislikes[i][0] < dislikes[i][1]
对于 dislikes[i] == dislikes[j] 不存在 i != j

样例

例 1：

输入：N = 4, dislikes = [[1,2],[1,3],[2,4]]
输出：true
解释：group1 [1,4], group2 [2,3]

例 2：

输入：N = 3, dislikes = [[1,2],[1,3],[2,3]]
输出：false

例 3：

输入：N = 5, dislikes = [[1,2],[2,3],[3,4],[4,5],[1,5]]
输出：false

bool possibleBipartition(int n, vector<vector<int>> &dislikes)
{
    map<int, set<int>>mapSet;
    vector<bool> row1(n + 1, false);
    vector<bool> row2(n + 1, false);
    vector<bool> visited(n + 1, false);
    vector<int> vec1;
    vector<int> vec2;
    bool bRow1 = false;
    for (int i = 0; i < dislikes.size(); i++)
    {
        if (vec1.empty())
        {
            vec1.push_back(dislikes[i][0]);
            row1[dislikes[i][0]] = true;
            visited[dislikes[i][0]] = true;
            bRow1 = true;
        }
        mapSet[dislikes[i][0]].insert(dislikes[i][1]);
        mapSet[dislikes[i][1]].insert(dislikes[i][0]);
    }

    while (!vec1.empty())
    {
        if (true == bRow1)
        {
            for (int i = 0; i < vec1.size(); i++)
            {
                set<int> tmpSet = mapSet[vec1[i]];

                //把值分类到row2
                for (auto it : tmpSet)
                {
                    if (row1[it] == true)
                    {
                        return false;
                    }
                    if (visited[it] == true)
                    {
                        continue;
                    }
                    vec2.push_back(it);
                    row2[it] = true;
                    visited[it] = true;
                }    
            }
            bRow1 = false;
        }
        else if (false == bRow1)
        {
            for (int i = 0; i < vec1.size(); i++)
            {
                set<int> tmpSet = mapSet[vec1[i]];
                //把值分类到row1
                for (auto it : tmpSet)
                {
                    if (row2[it] == true)
                    {
                        return false;
                    }
                    if (visited[it] == true)
                    {
                        continue;
                    }
                    visited[it] = true;
                    vec2.push_back(it);
                    row1[it] = true;
                }
            }
            bRow1 = true;
        }
        
        vec1 = vec2;
        vec2.clear();
    }
    return true;
}