洛谷P2024 [NOI2001]食物链

最新推荐文章于 2022-03-27 22:50:36 发布

原创最新推荐文章于 2022-03-27 22:50:36 发布 · 432 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#带权并查集

并查集专栏收录该内容

7 篇文章

订阅专栏

本文深入解析了NOI2001竞赛题“食物链”，采用带权并查集算法解决动物间的食物关系判断问题，通过维护fa和d数组，实现了对动物类型和关系的有效追踪。

题目：[NOI2001]食物链

思路：

带权并查集。

在维护fa的同时维护d，代表是ABC中的哪种动物。

代码：

#include<bits/stdc++.h>
using namespace std;

#define maxn 50000

int n,m;
int fa[maxn+5]={0};
int d[maxn+5]={0};

int find(int x) {
	if(fa[x]==0) return x;
	int y=fa[x];
	fa[x]=find(fa[x]);
	d[x]=(d[x]+d[y])%3;
	return fa[x];
}

void Merge(int x,int y,int fa1,int fa2,bool flg){
	fa[fa1]=fa2;
	d[fa1]=(3+d[y]-d[x]+flg)%3;
}

int main(){
	scanf("%d%d",&n,&m);
	int ans=0;
	for(int t=1;t<=m;t++){
		int opr,x,y;
		scanf("%d%d%d",&opr,&x,&y);
		if(opr==2&&x==y) continue;
		if(x>n||y>n) continue;
		int fa1=find(x),fa2=find(y);
		if(opr==1) {
			if(fa1==fa2&&d[x]==d[y]) ans++;
			else if(fa1!=fa2) {
				Merge(x,y,fa1,fa2,0);
				ans++;
			}
		} else {
			if(fa1==fa2&&d[x]==(d[y]+1)%3) ans++;
			else if(fa1!=fa2){
				Merge(x,y,fa1,fa2,1);
				ans++;
			}
		}
	}
	printf("%d",m-ans);
	
	return 0;
}