2020牛客多校训练3 G Operating on a Graph（并查集+链式结构）

.

G题

题意：每一次q，只要a团b团相邻，就将其合并

开一个vector维护每个团内的点和相邻的点的信息，用并查集维护每个点的所属团，每次将$o$统领与和它相邻的团合并

做法就对于每一个跟$o$团相邻的点（属于$o$团的点直接跳过），遍历所有属于这个点的祖先的边将其团内所有的点和$o$团相连即可

#include<bits/stdc++.h>
using namespace std;
const int maxm=8e5+5;
vector<int>e[maxm];
int n,m;
int f[maxm];
int find(int x){
	return f[x]==x?x:f[x]=find(f[x]);
}
void hb(vector<int>&x,vector<int>&y){
	if(x.size()<y.size())swap(x,y);
	for(int u:y)x.push_back(u);
}
int main(){
	int t;
	cin>>t;
	while(t--){
		scanf("%d%d",&n,&m);
		for(int i=0;i<n;i++){
			f[i]=i;
			e[i].clear();
		}
		for(int i=1;i<=m;i++){
			int u,v;
			scanf("%d%d",&u,&v);
			e[u].push_back(v);
			e[v].push_back(u);
		}
		int q;
		scanf("%d",&q);
		int o;
		while(q--){
			scanf("%d",&o);
			if(find(o)!=o)continue;
			vector<int>now=e[o];
			e[o].clear();
			for(auto v:now){
				int fv=find(v);
				if(fv==o)continue;
				hb(e[o],e[fv]);
				f[fv]=o;
			}
		}
		for(int i=0;i<n;i++){
			printf("%d ",find(i));
		}
		puts("");
	}
	return 0;
}