染色法判定二分图 模板 题解《挑战程序设计竞赛》

地址 https://www.acwing.com/problem/content/862/

给定一个n个点m条边的无向图，图中可能存在重边和自环。

请你判断这个图是否是二分图。

输入格式

第一行包含两个整数n和m。

接下来m行，每行包含两个整数u和v，表示点u和点v之间存在一条边。

输出格式

如果给定图是二分图，则输出“Yes”，否则输出“No”。

数据范围

1≤n,m≤105

输入样例：

4 4
1 3
1 4
2 3
2 4

输出样例：

Yes

基本模板

遍历图 然后随机分配颜色 与该点相邻的点必须是另一种颜色

模板代码

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

using namespace std;

const int N = 100010, M = 200010;

int n, m;
int h[N], e[M], ne[M], idx;
int color[N];

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

bool 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])
        {
            if (!dfs(j, 3 - c)) return false;
        }
        else if (color[j] == c) return false;
    }

    return true;
}

int main()
{
    scanf("%d%d", &n, &m);

    memset(h, -1, sizeof h);

    while (m -- )
    {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b), add(b, a);
    }

    bool flag = true;
    for (int i = 1; i <= n; i ++ )
        if (!color[i])
        {
            if (!dfs(i, 1))
            {
                flag = false;
                break;
            }
        }

    if (flag) puts("Yes");
    else puts("No");

    return 0;
}

自己代码

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>


using namespace std;

/*
输入格式
第一行包含两个整数n和m。

接下来m行，每行包含两个整数u和v，表示点u和点v之间存在一条边。

输出格式
如果给定图是二分图，则输出“Yes”，否则输出“No”。

数据范围
1≤n,m≤105
输入样例：
7 7
1 3
1 4
2 3
2 4
5 6
6 7
5 7
输出样例：
No
*/

const int N = 100010, M = 200010;

int n, m;

vector<vector<int>> uv(100010);

vector<int> colorr(100010);

string ret;

void dfs(int idx,int color) {
    if (idx > n || ret == "No" ) {
        return;
    }

    colorr[idx] = color;
    
    //变色
    color = color % 2 + 1;
    for (int i = 0; i < uv[idx].size(); i++) {
        //与该点相邻的点 都染上其他颜色
        int coloridx = uv[idx][i];
        //颜色冲突 则直接返回
        if (colorr[coloridx] != 0 && colorr[coloridx] != color)
        {
            ret = "No";
            return;
        }
        if(ret.empty() && colorr[coloridx] == 0)
            dfs(coloridx, color);
    }


    return;
}


int main()
{
    cin >> n >> m;

    for (int i = 0; i < m; i++) {
        int a; int b;
        cin >> a >> b;
        uv[a].push_back(b);
        uv[b].push_back(a);
    }
    /*for(int i= 1;i < n;i++)
        dfs(i,1);*/

    for (int i = 1; i <= n; i++) {
        if (ret == "No") break;
        if (colorr[i] != 0)  continue;
        dfs(i, 1);
    }

    if (ret.empty())  ret = "Yes";
    cout << ret << endl;


    return 0;
}