poj2492-A Bug‘s Life（kuangbin专题之权值并查集）

传送门

题意

一种虫子只有两种性别，现在将给出多组虫子间的互动关系（默认这些虫子都是异性恋），判断其中是否存在同性恋的虫子。

首先默认所有点与其父节点的距离为 $1$ ，那么对于在同一棵树上的结点，异性恋一定是这两个结点间的距离为奇数，同性恋一定是这两个结点间的距离为偶数，可以将该要求通过两结点到根的距离之差判断。

容易发现，如果两个结点不在同一棵树上时没有意义（此时一定不会发生同性恋事件），因此给出的每一组互动关系，当两点处于同一棵树上时为查询操作，不在同一棵树上时为加边更新操作。

在加边更新操作中，为了保证所选的两个结点为异性恋，此处假设原 $x$ 到其原祖先 $roo{t}_x$ 的距离为 $di{s}_x$，原 $y$ 到其原祖先 $roo{t}_y$ 的距离为 $di{s}_y$，若选择 $roo{t}_y$ 为新的根，则 $roo{t}_x$ 与 $roo{t}_y$ 之间需要建一条新的边 $dis$ 。此时两节点到新祖先 $roo{t}_y$ 的距离分别为 $di{s}_x+dis$ 和 $di{s}_y$ ，且这两个距离的奇偶性不同，即 $(di{s}_x+dis-di{s}_y)\%2==1$ ，由此又能转化为 $(di{s}_x+dis-di{s}_y-1)\%2==0$ ，此时可直接偷懒取 $dis=di{s}_y-di{s}_x+1$ ，则新建边的权值为此。

参考代码

#include <iostream>
#include <stdlib.h>
#define int long long
#define endl "\n"
using namespace std;

const int N=2050;
int fa[N],dis[N];//父节点 距离

int find(int x){
	if(x==fa[x])
		return x;
	
	int temp=fa[x];
	fa[x]=find(fa[x]);
	
	//处理 
	dis[x]+=dis[temp];

	return fa[x];
}

/* 
不在一棵树上没有影响，因为性别不确定
只考虑在同一棵树上与根节点的距离 
对于每一次操作 不在同一颗树上就更新
在同一颗树上查询是否同性（即与根节点距离之差是否为偶数 
构造新建的边长为 dis[x]-dis[y]+1
*/

void add(int a,int b){
	int aa=find(a),bb=find(b);
	fa[aa]=bb;
	
	dis[aa]=abs(dis[a]-dis[b]+1);
	//处理
}

signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int T;
	cin>>T;
	for(int t=1;t<=T;t++){
		int n,m,a,b;
		cin>>n>>m;
		for(int i=1;i<=n;i++){
			fa[i]=i;
			dis[i]=0;
		}
		
		int flag=0;
		for(int i=1;i<=m;i++){
			cin>>a>>b;
			if(find(a)!=find(b))
				add(a,b);
			else{//距离差是否为偶数 
				if(abs(dis[a]-dis[b])%2==0)
					flag=1;
			}
		}
		
//		for(int j=1;j<=n;j++){
//			cout<<find(j)<<" "<<dis[j]<<"\n";
//		}
//		cout<<endl;
		
		cout<<"Scenario #"<<t<<":\n";
		if(flag)
			cout<<"Suspicious bugs found!\n";
		else
			cout<<"No suspicious bugs found!\n";
		if(t!=T)
			cout<<"\n";
	}
	
	return 0;
}