洛谷 P3144 [USACO16OPEN]关闭农场Closing the Farm_Silver

P3144 [USACO16OPEN]关闭农场Closing the Farm_Silver

题目描述

Farmer John and his cows are planning to leave town for a long vacation, and so FJ wants to temporarily close down his farm to save money in the meantime.

The farm consists of NN barns connected with MM bidirectional paths between some pairs of barns (1 leq N, M leq 30001≤N,M≤3000). To shut the farm down, FJ plans to close one barn at a time. When a barn closes, all paths adjacent to that barn also close, and can no longer be used.

FJ is interested in knowing at each point in time (initially, and after each closing) whether his farm is "fully connected" -- meaning that it is possible to travel from any open barn to any other open barn along an appropriate series of paths. Since FJ's farm is initially in somewhat in a state of disrepair, it may not even start out fully connected.

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

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

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

输入输出格式

输入格式：

 

The first line of input contains NN and MM. The next MM lines each describe a

path in terms of the pair of barns it connects (barns are conveniently numbered

1 ldots N1…N). The final NN lines give a permutation of 1 ldots N1…N

describing the order in which the barns will be closed.

 

输出格式：

 

The output consists of NN lines, each containing "YES" or "NO". The first line

indicates whether the initial farm is fully connected, and line i+1i+1 indicates

whether the farm is fully connected after the iith closing.

 

输入输出样例

输入样例#1：

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

输出样例#1：

YES
NO
YES
YES
思路：正难则反，倒着进行合并，并查集维护，每次判断一下，是否只有一个根节点即可。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 3001
using namespace std;
int n,m,tot;
int fa[MAXN],opt[MAXN],vis[MAXN],ans[MAXN];
int to[MAXN],net[MAXN],head[MAXN],from[MAXN];
void add(int u,int v){
    to[++tot]=v;from[tot]=u;net[tot]=head[u];head[u]=tot;
}
int find(int x){
    if(fa[x]==x)    return x;
    else return fa[x]=find(fa[x]);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
    }
    for(int i=1;i<=n;i++){
        scanf("%d",&opt[i]);
        fa[i]=i;
    }
    for(int i=n;i>=1;i--){
        vis[opt[i]]=1;
        for(int j=head[opt[i]];j;j=net[j])
            if(vis[from[j]]&&vis[to[j]]){
                int dx=find(from[j]);
                int dy=find(to[j]);
                if(dx==dy)    continue;
                fa[dy]=dx;
            }
        int num=0;
        for(int j=1;j<=n;j++)
            if(vis[j]&&fa[j]==j)    num++;
        if(num==1)    ans[i]=1;
    }
    cout<<"YES"<<endl;
    for(int i=2;i<=n;i++)
        if(ans[i]==1)    cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
}