网址
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";
}