中等-886. 可能的二分法

第一次做这种类型的题目，看来我还是有很多需要学习的

给定一组 n 人（编号为 1, 2, ..., n）， 我们想把每个人分进任意大小的两组。每个人都可能不喜欢其他人，那么他们不应该属于同一组。

给定整数 n 和数组 dislikes ，其中 dislikes[i] = [ai, bi] ，表示不允许将编号为 ai 和  bi的人归入同一组。当可以用这种方法将所有人分进两组时，返回 true；否则返回 false。

 

示例 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
 

提示：

1 <= n <= 2000
0 <= dislikes.length <= 104
dislikes[i].length == 2
1 <= dislikes[i][j] <= n
ai < bi
dislikes 中每一组都 不同
 

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/possible-bipartition
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

/**
 * @param {number} n
 * @param {number[][]} dislikes
 * @return {boolean}
 */
var possibleBipartition = function(n, dislikes) {
    //统计i不喜欢的人，生成字典
    const dsMap=[]
    const colorMap=[]//0表示没有分类
    for(let i=0;i<=n;i++){
        dsMap[i]=[]
        colorMap[i]=0
    }
    dislikes.forEach(function(p){
        dsMap[p[0]].push(p[1])
        dsMap[p[1]].push(p[0])
    })
    //对有关联的人分成两类，一直寻找关联到结束
    
    for(let i=1;i<=n;i++){
        if(colorMap[i]===0){
            colorMap[i]=1;//给i编色为1，他不喜欢的编色为2
            const preList=[i]
            while(preList.length>0){
                let curI=preList.shift()
                for(let k=0;k<dsMap[curI].length;k++){
                    const dI=dsMap[curI][k]
                    //如果已经编过色了，判断是否同色，同色的话就false
                    if(colorMap[dI]>0&&colorMap[dI]===colorMap[curI]){
                        return false
                    }
                    //如果没编过色，就进行编色
                    if(colorMap[dI]===0){
                       colorMap[dI]=colorMap[curI]^3
                       preList.push(dI)
                    }
                }
            }
            
        }
        
    }
    return true
};