P3144 [USACO16OPEN]Closing the Farm S

FJ 和他的奶牛们正在计划离开小镇做一次长的旅行，同时 FJ 想临时地关掉他的农场以节省一些金钱。

这个农场一共有被用 M 条双向道路连接的 N 个谷仓（1≤N,M≤3000）。为了关闭整个农场，FJ 计划每一次关闭掉一个谷仓。当一个谷仓被关闭了，所有的连接到这个谷仓的道路都会被关闭，而且再也不能够被使用。

FJ 现在正感兴趣于知道在每一个时间（这里的“时间”指在每一次关闭谷仓之前的时间）时他的农场是否是“全连通的”——也就是说从任意的一个开着的谷仓开始，能够到达另外的一个谷仓。注意自从某一个时间之后，可能整个农场都开始不会是“全连通的”。

输入格式

输入第一行两个整数 N,M。

接下来 M 行，每行两个整数 u,v（1 ≤u,v≤N），描述一条连接 u,v 两个农场的路。

最后 N 行每行一个整数，表示第 ii 个被关闭的农场编号。

输出格式

输出 N 行，每行包含 YES 或 NO，表示某个时刻农场是否是全连通的。

第一行输出最初的状态，第 i行（2≤i≤N）输出第 i−1 个农场被关闭后的状态。

输入输出样例

输入 #1复制

4 3
1 2
2 3
3 4
3
4
1
2

输出 #1复制

YES
NO
YES
YES

分析,并查集合并容易,拆分很难,所以这道题我们反过来想,从末尾开始,一个一个点开通,看这个时候是否联通,然后,输出

#include<cstdio>
#include<cstring>
using namespace std;
int n,m,fa[3001],x[3001],y[3001],order[3001],ss[3001],ans[3001],w;

int find(int x)
{
        return x==fa[x] ? x:fa[x]=find(fa[x]);
}
void merg(int u,int v)
{
    fa[find(u)]= find(v);
}
int main()
{
    memset(ss,0,sizeof(ss));
    scanf("%d%d",&n,&m);
    for(int j=1;j<=n;j++) {
        fa[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\n");
        else printf("NO\n");
    printf("YES");
    return 0;
}