并查集经典:食物链

链接:登录—专业IT笔试面试备考平台_牛客网

题目描述:
并查集经典:食物链

并查集经典:食物链 

 直接上思路:

在这个问题中,A B C三种生物的地位是相同的,所以如果1 与 2是一个物种,1是A,2是A与1是B,2是B并没有区别,所以我们可以建立一个三倍生物数量的数组,然后对三种情况进行讨论,好处是情况绝对全面。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll n,k,fa[150010];
int find(int x){
	return fa[x]==x?x:fa[x]=find(fa[x]);
}
void hebin(int x,int y){
	fa[find(x)]=find(y);
}
int main()
{ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	cin>>n>>k;
	for(int i=1;i<=3*n;++i) fa[i]=i;
	ll cnt=0;
	for(int i=1;i<=k;++i){
		int op,x,y;
		cin>>op>>x>>y;
		if(x>n||y>n){//按题意取否;
			cnt++;
		    continue;
		}
   //x代表生物是A;
   //x+n代表生物是B;
   //x+2*n代表生物是C;
		else{
			if(op==1){
				if(find(x)==find(y+n)||find(x)==find(y+2*n))
				cnt++;
				else{
					hebin(x,y);
					hebin(x+n,y+n);
					hebin(x+2*n,y+2*n);
				}
			}
			else if(op==2){
				if(find(x)==find(y)||find(x)==find(y+2*n))
				cnt++;
				else{
					hebin(x,y+n);
					hebin(x+n,y+n*2);
					hebin(x+2*n,y);
				}
			}
		}
	}
	cout<<cnt<<endl;
	return 0;
}

上一篇:多态的注意事项


下一篇:2021 CCPC网络赛重赛