【问题描述】
C: City
Time Limit: 1 s Memory Limit: 128 MB
Problem Description
如果城市A和城市B互通,城市B和城市C互通,那么城市A和城市C也互通,A、B、C三个城市算一个聚集点。先已知有n个城市和m条道路,想求的是有几个聚集点?但小S觉得太简单了,由于战争原因,某些城市会被导弹销毁掉,与之相应的道路也变得不可用。之前已经被销毁的不会被复原。现给定每次销毁的城市顺序,求每次销毁后聚集点有多少个。
Input
第一行输入n和m,表示城市数量和道路数量(1≤n≤104,1≤m≤2n)
接下来m行,每行输入两个数ai和bi (1≤ai,bi≤n)。表示ai和bi直接有道路
第m+2行输入q,表示有q个城市会被销毁 (1≤q≤n)
接下来输入q个数,每行输入一个不重复的数,表示被销毁的城市
Output
输出一行q个数,每i个数表示第i个城市销毁后聚集点的数量
Sample Input
8 9
1 2
1 3
1 6
2 4
3 6
4 5
4 7
5 7
5 8
4
3 2 5 4
Sample Output
1 2 3 3
分析:先把炸毁的城市存起来,然后倒着建并差集
之前学了链式前向星,这里敲了下,勉强用在这题
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <string>
using namespace std;
const int N=1e4+100;
struct edge
{
int st;
int to;
int next;
}edge[4*N];
int head[4*N];
int fa[2*N];
int e[2*N];
int n, m, q;
int cnt;
void add(int u,int v)
{
edge[cnt].to=v;
edge[cnt].st=u;
edge[cnt].next=head[u];
head[u]=cnt++;
}
int find(int x)
{
return fa[x]==x?x:fa[x]=find(fa[x]);
}
void unite(int u,int v)
{
int fu=find(u),fv=find(v);
if(fu!=fv)
fa[fu]=fv;
}
int val[N];
int tt[N];
int main()
{
int u, v;
int x;
scanf("%d%d",&n,&m);
memset(head,-1,sizeof(head));
memset(e,0,sizeof(e));
for(int i=1;i<=n;i++) fa[i]=i;
cnt=0;
for(int i=0;i<m;i++)
{
scanf("%d%d",&u,&v);
add(u,v);
add(v,u);//双向存储
}
scanf("%d",&q);
for(int i=0;i<q;i++)
{
scanf("%d",&x);
e[x]=1;
tt[i]=x;
}
for(int w=0;w<cnt;w++)
{
int v=edge[w].to;
int u=edge[w].st;
if(!e[u]&&!e[v])
{
unite(u,v);
}
}
int nct=0;
int ans=0;
for(int j=1;j<=n;j++)
{
if(fa[j]==j&&!e[j]) nct++;
}
val[ans++]=nct;
for(int i=q-1;i>0;i--)
{
int u=tt[i];//城市
e[u]=0;
for(int k=head[u];~k;k=edge[k].next)
{
int v=edge[k].to;
if(!e[v])
unite(u,v);
}
//
nct=0;
for(int j=1;j<=n;j++)
{
if(fa[j]==j&&!e[j]) nct++;
}
val[ans++]=nct;
}
for(int i=ans-1;i>=0;i--)
{
if(i==0)
printf("%d\n",val[i]);
else
printf("%d ",val[i]);
}
return 0;
}