P3144 [USACO16OPEN]Closing the Farm S

算法

并查集+逆序

思路

首先读入相连的点，但这里不能直接合并建立并查集，因为并查集没有Ctrl+Z操作(就是无法分离两个已经合并的集合)，所以我们要先存起来，等所有的询问都读入之后，倒着进行操作。

我们考虑怎样倒着操作：

首先，读入数据，把所有的数据都存起来，其中x[i],y[i]表示第i次读入的关系，order[i]表示第i次读入的数是多少，ss[i]表示i是否在并查集里面，如果存在，则为0，不存在则为1。

接着，倒着处理读入的询问。从第i=n次开始，把与点order[i]有关的边读入，合并并查集，之后在把和该点有关的所有的可加入的边都加入并查集以后，判断并查集中集合的个数，并记录在ans[i]中，然后i--，重复以上步骤。

最后，从1开始到n-1，判断ans是否为1，如果为1，说明所有的点都是联通的（只有一个并查集），输出YES，否则输出NO，第n次询问的时候，所有的点都已经从并查集删除，因此一定是联通的，输出YES。

代码

#include<cstdio>
#include<cstring>
using namespace std;
int n,m,g[3001],x[3001],y[3001],order[3001],ss[3001],ans[3001],w;
//x[i],y[i]表示第i次读入的关系，order[i]表示第i次读入的数是多少，ss[i]表示i是否在并查集里面，如果存在，则为0，不存在则为1
//g[i]存储并查集，ans[i]存储第i次询问时并查集中有多少个集合。
int find(int u)
{
    if(g[u]!=u)g[u]=find(g[u]);
    return g[u];
}
void merg(int u,int v)
{
    u=find(u);
    v=find(v);
    if(u==v)return;
    g[u]=v;
}
int main()
{
    memset(ss,0,sizeof(ss));
    scanf("%d%d",&n,&m);
    for(int j=1;j<=n;j++)g[j]=j;
    for(int i=1;i<=m;i++)
        scanf("%d%d",&x[i],&y[i]);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&w);
        order[i]=w;
        ss[w]=1;//不在并查集内
    }
for(int i=n;i>0;i--)
    {
        ss[order[i]]=0;//由于倒序，此时它被建造
        for(int j=1;j<=m;j++)
            if(ss[x[j]]==0&&ss[y[j]]==0)//建造后可以合并
                merg(x[j],y[j]);
        ans[n]=0;
        for(int j=1;j<=n;j++)
            if(find(j)==j&&ss[j]==0)//为并查集的根（在并查集内且为自己祖先）
                ans[i]++;
    }
    for(int i=1;i<=n-1;i++)
        if(ans[i]==1)printf("YES
");//只有一个并查集
        else printf("NO
");//多个并查集不连通
    printf("YES");
    return 0;
}