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

洛谷 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 cc{
    int from,to;
}es[maxn];
int first[maxn],nxt[maxn];
int tot=0;
void build(int ff,int tt)
{
    es[++tot]=(cc){ff,tt};
    nxt[tot]=first[ff];
    first[ff]=tot;
}
bool vis[maxn];
int f[maxn];
int s[maxn];
int find(int w)
{
    if(w!=f[w])
    {
        f[w]=find(f[w]);
    }
    return f[w];
}
stack<int>ss;//存答案 
int main()
{
     int n,m;
     scanf("%d%d",&n,&m);
     for(int i=0;i<n;i++)//城市編號爲0~n-1,坑死人。。 
     {
        f[i]=i;
     }
     for(int i=1;i<=m;i++)
     {
        int x,y;
        scanf("%d%d",&x,&y);
        build(x,y);
        build(y,x);
     }
     int k;
     scanf("%d",&k);
     for(int i=1;i<=k;i++)
     {
        scanf("%d",&s[i]);
        vis[s[i]]=1;//將被摧毀的城市標記爲1 
     }
     int ans=n-k;
     for(int i=0;i<n;i++)
     {
        if(!vis[i])//如果當前城市未被摧毀 
        {
            for(int j=first[i];j;j=nxt[j])
            {
                int u=es[j].to;
                if(!vis[u]&&find(i)!=find(u))
                {
                    f[find(i)]=find(u);
                    ans--;
                }
            }
        }
     }
     ss.push(ans);
     for(int i=k;i>=1;i--)
     {
        vis[s[i]]=0;
        ans++;
        for(int j=first[s[i]];j;j=nxt[j])
        {
            int u=es[j].to;
            if(!vis[u]&&find(s[i])!=find(u))
            {
                f[find(s[i])]=find(u);
                ans--;
            }
        }
        ss.push(ans);
     }
     while(!ss.empty())
     {
        printf("%d\n",ss.top());
        ss.pop();
     }
     return 0;
}
發佈了131 篇原創文章 · 獲贊 254 · 訪問量 5萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章