[BZOJ4579][Usaco2016 Open]Closing the Farm（并查集）

题目描述

传送门

题解

和什么星球大战starwar那道题很像。。直接倒序上ufs就可以了。
动态维护联通块个数。
每条边只会被左右端点枚举两边，时间复杂度$O(2*m*\alpha(n))$

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define N 200005

int n,m,x,y,cnt;
int tot,point[N],nxt[N*2],v[N*2];
int f[N],q[N];
bool vis[N],ans[N];

void add(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
}
int find(int x)
{
    if (x==f[x]) return x;
    f[x]=find(f[x]);
    return f[x];
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;++i)
    {
        scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    for (int i=1;i<=n;++i) scanf("%d",&q[i]);
    for (int i=1;i<=n;++i) f[i]=i;
    for (int i=n;i>=1;--i)
    {
        cnt++;
        for (int j=point[q[i]];j;j=nxt[j])
            if (vis[v[j]])
            {
                int f1=find(q[i]),f2=find(v[j]);
                if (f1!=f2)
                {
                    f[f1]=f2;
                    cnt--;
                }
            }
        vis[q[i]]=1;
        if (cnt==1) ans[i]=1;
    }
    for (int i=1;i<=n;++i)
        if (ans[i]) puts("YES");
        else puts("NO");
}