bzoj4579: [Usaco2016 Open]Closing the Farm

#include<bits/stdc++.h>
#define rep(i,l,r) for(int i=l;i<=r;i++)
#define N 500000
using namespace std;
struct E{
    int to,next;
}e[N];
int n,m,out[N],f[N],tot,s[N],flag[N],head[N];
inline int read() {
     register int x=0,ch=getchar();
     while(ch<‘0‘||ch>‘9‘) ch=getchar();
     while(ch>=‘0‘&&ch<=‘9‘) x=x*10+ch-48,ch=getchar();
     return x;
}
int find(int x) {
     if(f[x]==x) return x;else return f[x]=find(f[x]);
}
inline void ins(int u,int v) {
     e[++tot].to=v; e[tot].next=head[u]; head[u]=tot;
}
int main () {
     register int a,b,cnt,x,y;
     scanf("%d%d",&n,&m);
     rep(i,1,m) {a=read(); b=read(); ins(a,b); ins(b,a);}
     rep(i,1,n) s[i]=read(),flag[s[i]]=1,f[i]=i; cnt=n;
     rep(i,1,n) if(!flag[i]){
          for(register int k=head[i];k;k=e[k].next) {
               if(!flag[e[k].to]) {
                    x=find(i); y=find(e[k].to);
                    if(x!=y) f[x]=y,--cnt;
               }
          }
     }
     for(register int i=n;i;i--) {
          flag[s[i]]=0;
          for(register int k=head[s[i]];k;k=e[k].next) if(!flag[e[k].to]) {
               x=find(s[i]); y=find(e[k].to);
               if(x!=y) f[x]=y,--cnt;
          }
          out[i]=(cnt==i);
     }
     rep(i,1,n) if(!out[i]) puts("NO");else puts("YES");
}