逆向并查集 HYSBZ1015星球大战starwar

 星球大战starwar

 HYSBZ - 1015 

 

　　很久以前，在一个遥远的星系，一个黑暗的帝国靠着它的超级武器统治者整个星系。某一天，凭着一个偶然的
机遇，一支反抗军摧毁了帝国的超级武器，并攻下了星系中几乎所有的星球。这些星球通过特殊的以太隧道互相直
接或间接地连接。 但好景不长，很快帝国又重新造出了他的超级武器。凭借这超级武器的力量，帝国开始有计划
地摧毁反抗军占领的星球。由于星球的不断被摧毁，两个星球之间的通讯通道也开始不可靠起来。现在，反抗军首
领交给你一个任务：给出原来两个星球之间的以太隧道连通情况以及帝国打击的星球顺序，以尽量快的速度求出每
一次打击之后反抗军占据的星球的连通快的个数。（如果两个星球可以通过现存的以太通道直接或间接地连通，则
这两个星球在同一个连通块中）。

Input

　　输入文件第一行包含两个整数，N (1  < =  N  < =  2M) 和M (1  < =  M  < =  200,000)，分别表示星球的
数目和以太隧道的数目。星球用 0 ~ N-1的整数编号。接下来的M行，每行包括两个整数X, Y，其中（0 < = X <> 
Y 表示星球x和星球y之间有“以太”隧道，可以直接通讯。接下来的一行为一个整数k，表示将遭受攻击的星球的
数目。接下来的k行，每行有一个整数，按照顺序列出了帝国军的攻击目标。这k个数互不相同，且都在0到n-1的范
围内。

Output

　　输出文件的第一行是开始时星球的连通块个数。接下来的N行，每行一个整数，表示经过该次打击后现存星球
的连通块个数。

Sample Input

8 13
0 1
1 6
6 5
5 0
0 6
1 2
2 3
3 4
4 5
7 1
7 2
7 6
3 6
5
1
6
3
5
7

Sample Output

1
1
1
2
3
3

来源： https://vjudge.net/problem/description/12267

 

第一次做这种题真的很难想，主要是不敢下手，在网上找了好多解析都是一言概之。还是觉得不要怕错，要自己按自己的思路写一下，发现其实还蛮简单的。

但感觉自己写的耗时有点多......

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstring>
#include<vector>
#include<stack>
#include<algorithm>
using namespace std;
#define N 400002
#define T(){printf("^^^ ");}
int bc[N], waitdel[N], vis[N], ans[N];;
vector<int>gra[N];
int find(int son)
{
    int fa = son;
    while(fa != bc[fa])
    {
        fa = bc[fa];
    }
    //
    int temp = son,k;
    while(fa != bc[temp])
    {
        k = bc[temp];
        bc[temp] = fa;
        temp = k;
    }
    return fa;
}
 
int main()
{
    int n,m,k,x,y;
    cin>>n>>m;
    for(int i=0; i<m; i++)
    {
        scanf("%d%d",&x,&y);
        gra[x].push_back(y);
        gra[y].push_back(x);
    }
    scanf("%d",&k);
    memset(vis,0,sizeof(vis));
    for(int i=0; i<k; i++)
    {
        scanf("%d",&waitdel[i]);
        vis[waitdel[i]] = 1;
    }
    for(int i=0; i<n; i++) bc[i] = i;
    int tot = n - k;
    for(int i=0; i<n; i++)
    {
        if(!vis[i])
        {
            for(int j=0; j<gra[i].size(); j++)
            {
                if(!vis[gra[i][j]])
                {
                    int x = find(i);
                    int y = find(gra[i][j]);
                    if(x != y)
                    {
                        tot --;
                        bc[y] = x;
                    }
                }
            }
        }
    }
    int top = 0;
    for(int i=k-1; i>=0; i--)
    {
        int k = waitdel[i];
        ans[top ++] = tot;
        tot ++;
        for(int j=0; j<gra[k].size(); j++)
        {
            if(!vis[gra[k][j]])
            {
                int x = find(k);
                int y = find(gra[k][j]);
                if(x != y)
                {
                    tot --;
                    bc[y] = x;
                }
            }
            vis[k] = 0;
        }
    }
    ans[top++] = tot;
    for(int i=top-1; i>=0; i--) printf("%d
",ans[i]);
}