【POJ 1182】食物链(并查集)

时间:2022-02-14 02:47:04

三种动物,A吃B,B吃C,C吃A。那么用并查集时,还要多一个x和根的关系,吃或者被吃或者同类。合并两个需要更新和祖先的关系。这个关系可以自己画一画。

#include<cstdio>
#define N 50005
struct lian
{
int fa;
int re;
} f[N];
int find(int x)
{
if(x==f[x].fa)return x;
int fa= f[x].fa;
f[x].fa=find(fa);
f[x].re=(f[x].re+f[fa].re)%;
return f[x].fa;
}
int ans,n,k;
int main()
{
scanf("%d%d",&n,&k);
for(int i=; i<=n; i++)
{
f[i].fa=i;
f[i].re=;
}
for(int i=; i<=k; i++)
{
int d,x,y;
scanf("%d%d%d" ,&d,&x,&y);
if(x>n||y>n)
{
ans++;
continue;
}
int fx=find(x);
int fy=find(y);
if(fx!=fy)
{
f[fx].fa=fy;
f[fx].re=(f[y].re-f[x].re+-d)%;
}
else
{
if(f[x].re!=f[y].re&&d==)
ans++;
else if(f[y].re!=(f[x].re+)%&&d==)
ans++;
}
}
printf("%d",ans);
}