POJ1182 食物链————并查集

最新推荐文章于 2021-10-01 14:22:11 发布

原创最新推荐文章于 2021-10-01 14:22:11 发布 · 550 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#struct

ACM 专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一个使用并查集算法解决复杂关系判断问题的例子。通过定义节点结构体及并查集的基本操作如查找和合并，实现了对输入指令的处理。具体地，该程序能够根据指令判断两个元素是否处于同一集合内且满足特定的关系条件。

#include<iostream> #include<string.h> typedef struct n { int f; int rel; }node; node data[50010]; int K,N; int findp(int x) { int tmp_p=data[x].f; if(tmp_p!=x) { int root=findp(tmp_p); data[x].f=root; data[x].rel=(data[x].rel+data[tmp_p].rel)%3; } return data[x].f; } void unions(int x,int y,int t) { int i=findp(x); int j=findp(y); data[i].f=j; if(t==1) data[i].rel=(data[y].rel-data[x].rel+6+1)%3; //对的 else data[i].rel=(data[y].rel-data[x].rel+6)%3; } int main() { int i,j,k,res=0,t; scanf("%d%d",&N,&K); for(i=0;i<=50000;i++) { data[i].f=i; data[i].rel=0; } for(i=0;i<K;i++) { scanf("%d%d%d",&t,&j,&k); if(j>N||k>N||(t==2&&j==k)) { res++; continue; } if(t==1) { if(findp(j)!=findp(k)) unions(j,k,0); else if(data[j].rel!=data[k].rel) { res++; } } else { if(findp(j)!=findp(k)) unions(j,k,1); else if((data[k].rel-data[j].rel+6+1)%3!=0) { res++; } } } printf("%d/n",res); return 0; }