洛谷P1197 [JSOI2008]星球大戰(並查集)

題目描述

很久以前,在一個遙遠的星系,一個黑暗的帝國靠着它的超級武器統治者整個星系。某一天,憑着一個偶然的機遇,一支反抗軍摧毀了帝國的超級武器,並攻下了星系中幾乎所有的星球。這些星球通過特殊的以太隧道互相直接或間接地連接。

但好景不長,很快帝國又重新造出了他的超級武器。憑藉這超級武器的力量,帝國開始有計劃地摧毀反抗軍佔領的星球。由於星球的不斷被摧毀,兩個星球之間的通訊通道也開始不可靠起來。現在,反抗軍首領交給你一個任務:給出原來兩個星球之間的以太隧道連通情況以及帝國打擊的星球順序,以儘量快的速度求出每一次打擊之後反抗軍佔據的星球的連通快的個數。(如果兩個星球可以通過現存的以太通道直接或間接地連通,則這兩個星球在同一個連通塊中)。

輸入輸出格式

輸入格式:
輸入文件第一行包含兩個整數,N (1 <= N <= 2M) 和M (1 <= M <= 200,000),分別表示星球的數目和以太隧道的數目。星球用0~N-1的整數編號。

接下來的M行,每行包括兩個整數X, Y,其中(0<=X<>Y < N),表示星球X和星球Y之間有以太隧道。注意所有的以太隧道都是雙向的。

接下來一行是一個整數K,表示帝國計劃打擊的星球個數。

接下來的K行每行一個整數X,滿足0<=X< N,表示帝國計劃打擊的星球編號。帝國總是按輸入的順序依次摧毀星球的。

輸出格式:
輸出文件的第一行是開始時星球的連通塊個數。

接下來的K行,每行一個整數,表示經過該次打擊後現存星球的連通塊個數。

輸入輸出樣例

輸入樣例#1:
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
輸出樣例#1:
1
1
1
2
3
3

思路:簡單的並查集,重點在於先存起來,然後倒序加邊找答案,存起來輸出。

代碼如下:

#include<iostream>
#include<cstdio>
#include<stack>
using namespace std;
const int maxn=400000+10;
struct dqs
{
    int f,t;
}hh[maxn];
int first[maxn],nxt[maxn],tj[maxn],fa[maxn],hm[maxn];//hm==毀滅 
bool used[maxn];
int tot=0;
void build(int f,int t)
{
    hh[++tot]=(dqs){f,t};
    nxt[tot]=first[f];
    first[f]=tot;
}
int find(int x)
{
    if(fa[x]==x) return x;
    else return fa[x]=find(fa[x]);
}
int main()
{
    int n,m,k;
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)    fa[i]=i;
    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        build(x,y);
        build(y,x);
    }
    scanf("%d",&k);
    for(int i=1;i<=k;i++)
    {
        scanf("%d",&hm[i]);
        used[hm[i]]=1;
    }
    int ans=n-k;
    for(int i=0;i<n;i++)
    {
        if(!used[i])
        {
            for(int j=first[i];j;j=nxt[j])
            {
                int u=hh[j].t;
                if(!used[u])
                {
                    int xx=find(i),yy=find(u);
                    if(xx!=yy)
                    {
                        fa[xx]=yy;
                        ans--;
                    }
                }
            }   
        }
    }
    tj[k+1]=ans;
    for(int i=k;i>=1;i--)
    {
        used[hm[i]]=0;
        ans++;
        for(int j=first[hm[i]];j;j=nxt[j])
        {
            int u=hh[j].t;
            if(!used[u])
            {
                int xx=find(hm[i]),yy=find(u);
                if(xx!=yy)
                {
                    fa[xx]=yy;
                    ans--;
                }   
            }
        }
        tj[i]=ans;
    }
    for(int i=1;i<=k+1;i++)
        printf("%d\n",tj[i]);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章