食物链 并查集

 

 

输入样例：

100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

输出样例：

3

难度：中等

时/空限制：1s / 64MB

总通过数：46032

总尝试数：99926

来源：《算法竞赛进阶指南》, 模板题 , NOI2001 , POJ1182 , kuangbin专题

算法标签

---

#include<bits/stdc++.h>
using namespace std;
int res=0;
const int N=100010;
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>>n>>m;
	for(int i=1;i<=n;i++) p[i]=i;
	while(m--)
	{
		int a,x,y;
		cin>>a>>x>>y;
		int px=find(x);
		int py=find(y);
		if(x>n||y>n) res++;
		else
		{
			if(a==1)
			{
				if(px==py)
					{if((d[x]-d[y])%3) res++;}
				else
				{
					p[px]=py;
					d[px]=d[y]-d[x];
				}
			}
			else
			{
				if(px==py)
					{if((d[x]-d[y]-1)%3) res++;}
				else
				{
					p[px]=py;
					d[px]=d[y]-d[x]+1;
				}
			}
		}
		
	}
	cout<<res<<endl;
}