您好,欢迎来到锐游网。
搜索
您的当前位置:首页2020牛客多校训练3 G Operating on a Graph(并查集+链式结构)

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

来源:锐游网

.

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

做法就对于每一个跟 o o o团相邻的点(属于 o o 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;
}

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- ryyc.cn 版权所有 湘ICP备2023022495号-3

违法及侵权请联系:TEL:199 1889 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务