其他分享
首页 > 其他分享> > 染色法判定是否是二分图

染色法判定是否是二分图

作者:互联网

算法思路:使用深度优先遍历遍历整张图,如果发现某个顶点已经染过色,并且和相邻顶点的颜色相同,则返回false,否则一直染下去,直到遍历完整张图。

代码实现:

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
const int N = 100010, M = 2 * N;

int h[N], e[M], ne[M], idx;
int color[N];
int n, m;
bool flag = true;

void add(int a, int b)
{
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx++;
}

void dfs(int u, int c)
{
    color[u] = c;
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j = e[i];
        if(!color[j])
        {
            dfs(j, 3-c);
        }
        else
        {
            if(color[j] == c)
            {
                flag = false;
                break;
            }
        }
    }
}

int main()
{
    memset(h, -1,sizeof h);
    cin >> n >> m;
    while(m--)
    {
        int u, v;
        cin >> u >> v;
        add(u, v);
        add(v, u);
    }
    for(int i=1;i<=n;i++)
    {
        // 使用for循环的原因是图可能不联通
        if(!color[i]) dfs(i, 2); // 只有没染过色才能染,染1或者染2都行
    }
    
    if(flag) puts("Yes");
    else puts("No");
    
    return 0;
}

代码来自ACWing(https://www.acwing.com/)。

标签:二分,idx,int,是否是,ne,color,add,染色法,include
来源: https://www.cnblogs.com/cvxer/p/16473419.html