星球大战-C++

星球大战

题目描述
很久以前,在一个遥远的星系,一个黑暗的帝国靠着它的超级武器统治着整个星系。某一天,凭着一个偶然的机遇,一支反抗军摧毁了帝国的超级武器,并攻下了星系中几乎所有的星球。这些星球通过特殊的以太隧道互相直接或间接地连接。
但好景不长,很快帝国又重新造出了他的超级武器。凭借这超级武器的力量,帝国开始有计划地摧毁反抗军占领的星球。由于星球的不断被摧毁,两个星球之间的通讯通道也开始不可靠起来。
现在,反抗军首领交给你一个任务:给出原来两个星球之间的以太隧道连通情况以及帝国打击的星球顺序,以尽量快的速度求出每一次打击之后反抗军占据的星球的连通块的个数。(如果两个星球可以通过现存的以太通道直接或间接地连通,则这两个星球在同一个连通块中)。
输入格式 输入文件第一行包含两个整数,n,m 分别表示星球的数目和以太隧道的数目。星球用 0∼n−1的整数编号。
接下来的 m 行,每行包括两个整数 x,y,表示星球 x 和星球 y 之间有 “以太” 隧道,可以直接通讯。
接下来的一行为一个整数 k ,表示将遭受攻击的星球的数目。
接下来的 k 行,每行有一个整数,按照顺序列出了帝国军的攻击目标。这 k 个数互不相同,且都在 000 到 n−1 的范围内。
输出格式 第一行是开始时星球的连通块个数。接下来的 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

这题我们可以用一个逆向思维来做。就是先求出经过n个顶点后的连通块数,然后根据顶点的修复。再逐渐求出在哪个顶点被摧毁后的连通块数。
C++代码:

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,cnt; 
struct edge{
  int u,v,next;
}e[500000];
int h[400005];
bool ck[400005];
int head[400005];
int f[400005];
int ans[400005];
void add(int u, int v)
{
   e[cnt].u = u;
   e[cnt].v = v;
   e[cnt].next = head[u];
   head[u] = cnt; 
   cnt ++;
}
int getf(int u)
{
   if(u != f[u])
   {
    f[u] = getf(f[u]);
    return f[u];
   }
   return u;
}
int main()
{
   cin >> n >> m;
   for(int i = 0; i < n; i ++)
    {
      f[i] = i;
      head[i] = -1;
    }
   for(int i = 0; i < m; i ++)
   {
    int u,v;
    scanf("%d%d",&u,&v);
    add(u,v);
    add(v,u);
   }
   int k;
   cin >> k;
   for(int i = 1; i <= k; i ++)
   {
    int u;
    scanf("%d",&u);
    //要摧毁的节点用数组做标记 
    h[i] = u; 
    ck[u] = true;
   }
   //被摧毁前的节点数
   int toc = n - k;
   //先求出k个节点被摧毁后的连通块个数
    for(int i = 0; i < 2 * m; i ++)
    {
      if(ck[e[i].u] == false && ck[e[i].v] == false)
      {
          if(getf(e[i].u) != getf(e[i].v))
          {
            toc --;
            f[getf(e[i].u)] = f[getf(e[i].v)]; 
     }
  }
    }
     ans[k + 1] = toc;
    //修复 
    for(int j = k; j >= 1; j --)
    {
     int u = h[j];
     ck[u] = false;
     toc ++;
     for(int t = head[u]; t != -1; t = e[t].next)
     {
         if( ck[e[t].v ]== false && f[getf(u)] != f[getf(e[t].v)] )
         {
           toc --;
  f[getf(e[t].u)] = f[getf(e[t].v)];
      }
     }  
 ans[j] = toc;
    }
    for(int i = 1; i <= k + 1; i ++)
    cout << ans[i] << endl; 
   return 0;
}
发布了26 篇原创文章 · 获赞 0 · 访问量 1343

猜你喜欢

转载自blog.csdn.net/weixin_43846217/article/details/104198818
今日推荐