BZOJ 1015-星球大战starwar（并查集）

1015: [JSOI2008]星球大战starwar

Time Limit: 3 Sec   Memory Limit: 162 MB
Submit: 6321   Solved: 2939
[ 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

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

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

HINT

求每删一个点后的联通块数，一般求联通块直接上并查集，没什么可说的，

可是很容易想到，正着删点的话可能需要n^2的复杂度，不用说，肯定超时。。。

所以反着搞就好啦，删点等于加点，一开始先求出删完所有点的联通块数，在一点点加就好啦

但是一开始的写法有问题，怎么搞都wa，后来发现可能有自己只和自己联通的情况。。。。。

版本一（wa）：

#include<map>
#include<stack>
#include<queue>
#include<vector>
#include<math.h>
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long  ll;
#define  maxn 450000
vector<int>q[maxn];
int flag[maxn],a[maxn],parent[maxn],ans[maxn],used[maxn],flag1[maxn];
int find(int x)
{
    if(parent[x]==x)
        return x;
    return parent[x]=find(parent[x]);
}
int  main(void)
{
    int n,m,i,j,x,y,k;
    scanf("%d%d",&n,&m);
    for(i=0;i<n;i++)
        parent[i]=i;
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        if(flag1[x])
            continue;
        q[x].push_back(y);
        if(x==y)
        {
            flag1[x]=1;
            continue;
        }
        q[y].push_back(x);
    }
    scanf("%d",&k);
    for(i=1;i<=k;i++)
    {
        scanf("%d",&a[i]);
        flag[a[i]]=1;
    }
    for(i=0;i<n;i++)
    {
        if(flag[i])
            continue;
        for(j=0;j<q[i].size();j++)
        {
            int v=q[i][j];
            if(flag[v])
                continue;
            int t1=find(i),t2=find(v);
            if(t1!=t2)
                parent[t1]=t2;
        }
    }
    for(i=0;i<n;i++)
    {
        if(flag[i] || used[find(i)])
            continue;
        ans[k]++;
        used[find(i)]=1;
    }
    for(i=k;i>0;i--)
    {
        flag[a[i]]=0;
        ans[i-1]=ans[i];
        bool mark=0;
        for(j=0;j<q[a[i]].size();j++)
        {
            int v=q[a[i]][j];
            //printf("%d\n",flag[v]);
            if(flag[v]==1)
                continue;
            int t1=find(a[i]),t2=find(v);
            if(v==a[i] && q[a[i]].size()==1)
            {
                ans[i-1]++;
                continue;
            }
            if(t1!=t2)
            {
                parent[t1]=t2;
                if(mark)
                    ans[i-1]--;
                mark=1;
            }
        }
    }
    for(i=0;i<=k;i++)
        printf("%d\n",ans[i]);
    return 0;
}

后来发现了一个样例。。。

3 3

1 2

2 2

3 3

我竟然输出1 1 1.。。。。。好吧

版本二（ac）：

#include<map>
#include<stack>
#include<queue>
#include<vector>
#include<math.h>
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long  ll;
#define  maxn 450000
vector<int>q[maxn];
int flag[maxn],a[maxn],parent[maxn],ans[maxn],used[maxn],flag1[maxn];
int find(int x)
{
	if(parent[x]==x)
		return x;
	return parent[x]=find(parent[x]);
}
int  main(void)
{
	int n,m,i,j,x,y,k,sum;
	scanf("%d%d",&n,&m);
	for(i=0;i<n;i++)
		parent[i]=i;
	for(i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		q[x].push_back(y);
		q[y].push_back(x);
	}
	scanf("%d",&k);
	for(i=1;i<=k;i++)
	{
		scanf("%d",&a[i]);
		flag[a[i]]=1;
	}
	sum=n;
	for(i=0;i<n;i++)
	{
		if(flag[i])
			continue;
		for(j=0;j<q[i].size();j++)
		{
			int v=q[i][j];
			if(flag[v])
				continue;
			int t1=find(i),t2=find(v);
			if(t1!=t2)
			{
				parent[t1]=t2;
				sum--;
			}
		}
	}
	ans[k]=sum;
	for(i=k;i>0;i--)
	{
		flag[a[i]]=0;
		bool mark=0;
		for(j=0;j<q[a[i]].size();j++)
		{
			int v=q[a[i]][j];
			if(flag[v]==1)
				continue;
			int t1=find(a[i]),t2=find(v);
			if(t1!=t2)
			{
				parent[t1]=t2;
				sum--;
			}
		}
		ans[i-1]=sum;
	}
	for(i=0;i<=k;i++)
		printf("%d\n",ans[i]-i);
	return 0;
}