csu 1601 1601: War (并查集 kruskal)

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int vis[100000+100];
int  fa[100000+100];
int del[100000+100];
int ans[100000+100];
int n,m,q;
struct Edge
{
    int f,to;
};
Edge edge[100000+100];
int fin(int a)
{
    return fa[a]==a?a:fa[a]=fin(fa[a]);
}
int kru(int x,int y)
{
    int fx=fin(x);
    int fy=fin(y);
    if(fx!=fy)
    {
       fa[fx]=fy;
       return 1;
    }
    return 0;
}
int main()
{

    int i,j,k;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(i=0;i<=n;i++) fa[i]=i;
        memset(vis,0,sizeof(vis));

        for(i=1;i<=m;i++)
        {
            scanf("%d%d",&edge[i].f,&edge[i].to);
        }
        scanf("%d",&q);
        for(i=0;i<q;i++)
        {
            scanf("%d",&del[i]);
            vis[del[i]]=1;
        }
        int cnt=0;
        int tmp=n;
        for(i=1;i<=m;i++)
        {
            if(vis[i]==0)
            {
                if(kru(edge[i].f,edge[i].to)==1) tmp--;
            }
        }
        ans[cnt++]=tmp;
        for(i=q-1;i>0;i--)
        {
            j=del[i];
            if(kru(edge[j].f,edge[j].to)==1) tmp--;
            ans[cnt++]=tmp;
        }
        for(i=cnt-1;i>=0;i--)
        {
            printf("%d",ans[i]);
            if(i!=0) printf(" ");
            else printf("\n");
        }
    }
    return 0;
}