2020牛客多校三 G Operating on a Graph

https://ac.nowcoder.com/acm/contest/5668/G
题意:n个点m条边,每个点属于1个组,q次操作,每次把一个组相邻的组(有点相邻就组相邻)的所有点变成属于自己的组,问最后每个点属于哪个组。
思路:并查集维护点属于哪个组是没有疑问的,时间复杂度不太好分析,因为每个点最多有一次把其相邻的点染成自己的颜色,之后就永远和相邻点同色了,所以复杂度 O ( n + m ) O(n+m) O(n+m),直观的想复杂度很大,实际上很小,存在一个问题,用vector会炸空间,需要每次合并把小的合并到大的上才勉强不会炸空间,很坑,专门卡vector。用前向星较为简便。

#include 
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;
}

你可能感兴趣的:(染色)