https://ac.nowcoder.com/acm/contest/5668/G
题意:n个点m条边,每个点属于1个组,q次操作,每次把一个组相邻的组(有点相邻就组相邻)的所有点变成属于自己的组,问最后每个点属于哪个组。
思路:并查集维护点属于哪个组是没有疑问的,时间复杂度不太好分析,因为每个点最多有一次把其相邻的点染成自己的颜色,之后就永远和相邻点同色了,所以复杂度
O
(
n
+
m
)
O(n+m)
O(n+m),直观的想复杂度很大,实际上很小,存在一个问题,用vector会炸空间,需要每次合并把小的合并到大的上才勉强不会炸空间,很坑,专门卡vector。用前向星较为简便。
#include <bits/stdc++.h>
using namespace std;
const int maxn=800000+100;
int T,n,m,q;
int head[maxn],nxt[maxn*2],last[maxn],to[maxn*2],tot;
int fa[maxn];
int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
void AddEdge(int x,int y)
{
to[++tot]=y;
nxt[tot]=head[x];
head[x]=tot;
if(!last[x])last[x]=tot;
}
int main()
{
//freopen("input.in","r",stdin);
cin>>T;
while(T--)
{
cin>>n>>m;
int x,y;
tot=0;
for(int i=1;i<=n;i++)fa[i]=i,head[i]=last[i]=0;
for(int i=1;i<=m;i++)
{
scanf("%d%d",&x,&y);
x++,y++;
AddEdge(x,y);
AddEdge(y,x);
}
cin>>q;
while(q--)
{
scanf("%d",&x);
x++;
if(x!=find(x))continue;
head[0]=last[0]=0;
for(int i=head[x];i;i=nxt[i])
{
int v=to[i];
int fv=find(v);
if(fv==x)continue;
fa[fv]=x;
nxt[last[fv]]=head[0];
head[0]=head[fv];
if(!last[0])last[0]=last[fv];
}
head[x]=head[0];
last[x]=last[0];
}
for(int i=1;i<=n;i++)printf("%d ",find(i)-1);
putchar('\n');
}
return 0;
}