poj 1182 带权并查集经典

来源：互联网发布：mac chill 试色编辑：程序博客网时间：2024/06/05 20:57

点击打开链接

#include <iostream>#include <cstdio>#include <algorithm>using namespace std;const int M = 51000;int par[M],rank[M],n;// rank[x]={0,1,2} x和父节点关系 同类 吃 被吃; int find(int x){if(x!=par[x]){int t=par[x];par[x]=find(par[x]);//路径压缩rank[x]=(rank[x]+rank[t])%3; // 画个表格推一下咯~ }return par[x];}void Union(int x,int y,int d) // x->y 关系为 d {int fx=find(x);int fy=find(y);if(fx!=fy){par[fx]=fy; // fx->fy 关系？ rank[fx]=(3-rank[x]+d+rank[y])%3;//fx->x->y->fy      }}int main(){int n,k,d;cin>>n>>k;for(int i=1;i<=n;i++){par[i]=i;}int count=0;for(int i=0;i<k;i++){int x,y;scanf("%d%d%d",&d,&x,&y);if(x>n||x<1||y>n||y<1||(d==2&&x==y)){count++;continue;}int rx=find(x);int ry=find(y);if(rx==ry)  //两个元素在同一组 才可以可以判定之间的关系 。{ // x->root r[x] 则root-> x  3-r[x] //当前x->y关系? 是否为 d-1? //  x->root->y  (r[x]+3-r[y])%3if((rank[x]+3-rank[y])%3 != d-1){count++;continue;}  }else {Union(x,y,d-1); //x->y d-1}}cout<<count<<endl;return 0;}

0 0