AcWing 240. 食物链
原题链接
中等
作者:
cordis
,
2020-08-10 11:59:35
,
所有人可见
,
阅读 368
总结
1.通过其他节点到祖宗节点的路径之间的关系%3来表示3种动物之间的关系(环形)
2.路径的累加是在find函数里实现的
p[],d[];
a,b为输入的数
x,y为a、b的祖宗节点
如果 a和b>n 则为假。
当D==1 x==y (d[a]-d[b])%3 res++,a和b时同类,到祖宗节点的距离相等%3!=0,则为假。
当D==1 x!=y p[x]=y是x的祖宗节点为y 然后算d[x]的距离,因为两点到祖宗节点相等,d[x]=d[b]-d[a];
当D==2 x==y (d[a]-d[b]-1)%3 res++,a吃b所以大1(距离大1),如果%3后!=0 则为假。
当D==2 x!=y p[x]=y,d[x]=d[b]+1-d[a]所有的距离都跟x==y时判断里的相反。
C++ 代码
#include <iostream>
using namespace std;
const int N=1e5+10;
int p[N],d[N];
int find(int x)
{
if(p[x]!=x)
{
int t=find(p[x]);
d[x]+=d[p[x]];
p[x]=t;
}
return p[x];
}
int main ()
{
int n,m;
cin.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++) p[i]=i;
int res=0;
while(m--)
{
int t,a,b;
cin>>t>>a>>b;
int x=find(a),y=find(b);
if(a>n||b>n) res++;
else
{
if(t==1)
{
if(x==y&&(d[a]-d[b])%3) res++;
else if(x!=y)
{
p[x]=y;
d[x]=d[b]-d[a];
}
}
else
{
if(x==y&&(d[a]-d[b]-1)%3) res++;
else if(x!=y)
{
p[x]=y;
d[x]=d[b]+1-d[a];
}
}
}
}
cout<<res<<endl;
return 0;
}