食物链
-
核心思想：带权并查集
-
- 用距根节点和距离表示与根节点的关系
-
求距离
-
-
#include<iostream>using namespace std;const int N=50010;int n,m;int p[N],d[N];//找到祖宗节点(路径压缩) 并求出对应距离int find(int x){if(p[x]!=x){int u=p[x]; //保存旧父节点d[x] += d[u];p[x] = find(p[x]); //路径压缩 所有节点指向祖宗节点}return p[x];}int main(){cin>>n>>m;for(int i=1;i<=n;i++) p[i]=i; //初始化int res=0;while(m--){int t,x,y;cin>>t>>x>>y;if(x>n||y>n) res++; //超出范围else{int px=find(x),py=find(y); //找到xy的祖宗节点if(t==1){ //说明是同类 判断对错if(px==py && (d[x]-d[y]) % 3!=0) res++; //说明在一个并查集中,且xy不同类(距离不是3的倍数)else if (px != py){ //不在一个并查集p[px] = py; //将x的祖宗节点的父节点改成y的祖宗节点d[px] = d[y] - d[x]; //xy同类->路径长度推算d[y]-d[x]}}else{ //说明是异类 判断对错if(px==py && (d[x]-d[y]-1)%3!=0) res++; //说明在一个并查集中,且x不能吃y(d[x]!=d[y]+1+3*k) else if (px != py){p[px] = py; d[px] = d[y] +1 -d[x]; //x吃y->推算d[px]长度}}}}cout<<res;}```
