Acwing 860. 染色法判定二分图

网址

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

过完年刷题遇到了小problem ，这道题说是简单，其实我刚开始知道ne,e,h是什么意思。有点类似trie图，但是又不完全一样。

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e5+10,M=2e5+10;
int ne[M],e[M],h[M],col[M];
int n,m,idx;
void add(int a,int b)
{
    e[idx]=b;ne[idx]=h[a];h[a]=idx++;//这一步比较费力。ne是在枚举a的各个子节点，e保存了a的子节点的id，h[a]
    //则是子节点的位置，（坐标）。
    return;
}
bool dfs(int x,int c)
{
    col[x]=c;
    for(int i=h[x];i!=-1;i=ne[i])
    {
        if(!col[e[i]])
        {
            if(!dfs(e[i],3-c)) return false;
        }
        else if(col[e[i]]==c)
        return false;
    }
    return true;
}
int main(void)
{
    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(!col[i])
        if(!dfs(i,1)) 
        {
            flag=false;
            break;
        }
    }
    if(flag) cout<<"Yes";
    else
    cout<<"No";
}

然后我稍微改了一下，用了更简单的vector来存子节点

#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N=1e5+10,M=2e5+10;
int col[M];
vector<int> v1[N];
int n,m,idx;
void add(int a,int b)
{
    v1[a].push_back(b);
    return;
}
bool dfs(int x,int c)
{
    col[x]=c;
    for(int  i=0;i<v1[x].size();i++)
    {
        if(!col[v1[x][i]])
        {
            if(!dfs(v1[x][i],3-c)) return false;
        }
        else if(col[v1[x][i]]==c)
        return false;
    }
    return true;
}
int main(void)
{
    scanf("%d%d",&n,&m);
    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(!col[i])
        if(!dfs(i,1)) 
        {
            flag=false;
            break;
        }
    }
    if(flag) cout<<"Yes";
    else
    cout<<"No";
}