LeetCode 886. Possible Bipartition

最新推荐文章于 2021-01-02 21:36:07 发布

yc_cy1999

最新推荐文章于 2021-01-02 21:36:07 发布

阅读量121

点赞数

分类专栏： # 图文章标签： dfs leetcode 算法

本文链接：https://blog.csdn.net/yc_cy1999/article/details/105941511

版权

图专栏收录该内容

62 篇文章 0 订阅

订阅专栏

文章目录

题目描述
结果
知识点
实现
码前思考
代码实现
码后反思

题目描述

在这里插入图片描述

结果

在这里插入图片描述

知识点

图的遍历BFS+DFS，二分图

实现

码前思考

二分图题就是图顶点着色问题，就是在图DFS上增加一些额外的操作，模板还是DFS的！
注意非连通图的情况！

代码实现

//数组下标从1开始
class Solution {
private:
    const static int maxn = 2010;
    vector<int> adj[maxn];
    int color[maxn];
    bool flag;
public:
    bool possibleBipartition(int N, vector<vector<int>>& dislikes) {
        //初始化参数
        fill(color,color+maxn,-1);
        flag = true;

        //构建邻接表
        for(int i=0;i<dislikes.size();i++){
            int u = dislikes[i][0];
            int v = dislikes[i][1];
            adj[u].push_back(v);
            adj[v].push_back(u);
        }

        //开始进行DFS
        for(int i=1;i<=N;i++){
            if(flag == false){
                break;
            }else{
                if(color[i] == -1){
                     DFS(i,0);
                }
            }
        }

        return flag;
    }

    void DFS(int root,int c){
        if(flag == false){
            return;
        }else{
            if(color[root]!=-1){
                if(color[root] != c){
                    flag = false;
                    return;
                }else{
                    return;
                }
            }else{
                color[root] = c;
                for(int i=0;i<adj[root].size();i++){
                    DFS(adj[root][i],1-c);
                }
            }
        }
    }
};