poj 1182-食物链解题报告

#include<stdio.h>
#include<string.h>
#define N 50005
struct node
{
int p;
int r;
};
node f
;
int n;
void init()
{
int i;
for(i=1;i<=n;i++)
{
f[i].p=i;
f[i].r=0;
}
}
int find(int x)
{
if(f[x].p==x)
return x;
int temp=f[x].p;
f[x].p=find(temp);
f[x].r=(f[temp].r+f[x].r)%3;
return f[x].p;
}
void uni(int x,int y,int sx,int sy,int d)
{
f[sy].p=sx;
f[sy].r=(3+(d-1)+f[x].r-f[y].r)%3;
}
int main()
{
int d,a,b,m;
int i,j,k;
scanf("%d%d",&n,&m);
init();
int num=0;
while(m--)
{
scanf("%d%d%d",&d,&a,&b);
if(a>n||b>n)
{
num++;
continue;
}
int pa=find(a);
int pb=find(b);
if(pa!=pb)
uni(a,b,pa,pb,d);
else
{
if(d==1&&f[a].r!=f[b].r)
num++;
else if(d==2&&(f[b].r-f[a].r+3)%3!=1)
num++;
}
}
printf("%d\n",num);
return 0;
}