BZOJ1015 [JSOI2008]星球大战starwar——逆向思维并查集+路径压缩

1015: [JSOI2008]星球大战starwar

Time Limit: 3 Sec   Memory Limit: 162 MB
Submit: 5558   Solved: 2553
[ Submit][ Status][ Discuss]

Description

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

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行，每行一个整数，表示经过该次打击后现存星球
的连通块个数。

       

       简述一下题目，就是给一个无向图，每次删除1点极其上所连的边，求剩余连通块的数量。由于题目没有强制在线，因此我们可以进行离线处理，加快速度。求连通块，我们容易想到并查集，但是由题意知，本题含有一个删点的操作，因此不能直接使用。由于是离线，我们可以将删点改为加点，使用并查集处理。

       要注意的是，本题需要路径压缩优化，否则不能通过。

#include <bits/stdc++.h>
#include <iostream>


using namespace std;


vector <int> l[400005];
//stack<int> que;
int num[400005];
int fa[400005];
bool vis[400005];
int ans1[200005];
int find1(int a)
{
    if(fa[a]!=a)
        fa[a]=find1(fa[a]);
    return fa[a];
}

void union1(int a,int b)
{
    fa[find1(a)]=find1(b);
}

bool qu(int a,int b)
{
    return find1(a)==find1(b);
}

int main()
{
    int m,n,l1,r;
    cin>>n>>m;
    memset(vis,0,sizeof(vis));
    for(int i=0;i<=n;i++)
        fa[i]=i;
    for(int i=1;i<=m;i++){
        scanf("%d %d",&l1,&r);
        l[l1].push_back(r);
        l[r].push_back(l1);
    }
    int q;
    cin>>q;
    for(int i=1;i<=q;i++)
    {
        int vi;
        scanf("%d",&num[i]);
        vis[num[i]]=1;
    }
    int ans=0;
    for(int i=0;i<n;i++)
    {
        if(!vis[i])
        {
            int si=l[i].size();
            for(int j=0;j<si;j++)
            {
                if(!vis[l[i][j]])  //important
                union1(i,l[i][j]);
            }
        }
    }
    for(int i=0;i<n;i++)
    {
        if(!vis[i]&&fa[i]==i)
            ans++;
    }
    //int ssi=que.size();
    for(int i=q;i>0;i--)
    {
        ans1[i]=ans;
        int u=num[i];
        vis[u]=false;
        ans++;
        int si=l[u].size();
        for(int j=0;j<si;j++)
        {
            if(!vis[l[u][j]])
            if(!qu(u,l[u][j])){
                union1(u,l[u][j]);
                ans--;
            }
        }
    }
    ans1[0]=ans;
    for(int i=0;i<=q;i++)
        printf("%d\n",ans1[i]);
    //fclose(stdin);
    //fclose(stdout);

    //cout << "Hello world!" << endl;
    return 0;
}