[BZOJ1015][JSOI2008]星球大战starwar

原题地址

倒着做并查集即可.

AC code:

#include <cstdio>
#include <vector>
using namespace std;
const int N=1000010;
int  n,m,o,cnt;
int  f[N],star[N],ans[N];
bool tag[N];
vector<int> G[N];

int find(int x){
    if(f[x]!=x) f[x]=find(f[x]);
    return f[x];
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++) f[i]=i;
    for(int i=1;i<=m;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    scanf("%d",&o);
    for(int i=1;i<=o;i++){
        scanf("%d",&star[i]);
        tag[star[i]]=1;
    }
    for(int i=0;i<n;i++){
        if(tag[i]) continue;
        for(int j=0;j<(int)G[i].size();j++){
            int v=G[i][j];
            if(tag[v]) continue;
            f[find(i)]=find(v);
        }
    }
    for(int i=0;i<n;i++){
        if(f[i]==i&&(!tag[i])) cnt++;
    }
    ans[o+1]=cnt;
    for(int i=o;i>=1;i--){
        cnt++;
        tag[star[i]]=0;
        for(int j=0;j<(int)G[star[i]].size();j++){
            int v=G[star[i]][j];
            if(tag[v]) continue;
            if(find(star[i])!=find(v)) cnt--;
            f[find(star[i])]=find(v);
        }
        ans[i]=cnt;
    }
    for(int i=1;i<=o+1;i++) printf("%d\n",ans[i]);


    return 0;
}