蓝桥杯.修改数组(并查集)

Question:

Solve:

数据量还是蛮大的，直接暴力1s跑不完，主要的时间都花费在读入一个数，查找没有出现过的第一个数上，考虑利用并查集优化这个过程

将出现过的连续数字都构造到一棵树里，然后将每一个数的父节点直接标记为与它连续的且前面出现过的最大数

那么在之后读入数据的时候如果这个数已经出现过，就可以直接找它的父节点，将这个数修改为父节点的数字加一输出

 eg：

在读完蓝色数据的时候，我构造出如图所示的两棵树，当再读入5的时候，就可以直接查找它的父节点为7，将5改为7+1=8

 但是两个问题：

1.树不是直接就构造成这样的，这涉及到了并查集的路径压缩

2.如果在上述例子中我再读入一个3，那么两棵树就会合并，合并成一个所有元素的父节点都是8的样子，所以在读入元素修改过后，还要判断修改数的前后数字是否出现过，也就是看树能不能合并

Code:

#include <bits/stdc++.h>
using namespace std;
#define MAXN 1000000
int fat[MAXN+1];  //并查集的父节点
bool vis[MAXN+1]; //判断每一个数是否出现过
//父节点查找
int find(int x)
{
  if(fat[x] == x) return x;
  //注意在不断查找根节点的过程中要路径压缩
  else return fat[x] = find(fat[x]);
}

int main(void)
{
  memset(vis,false,sizeof(vis));
  //并查集初始化
  for(int i = 1; i <= MAXN; i++) fat[i] = i;
  
  int n,x;
  cin >>n;
  for(int i = 1; i <= n; i++){
    cin >>x;
    //如果数字出现过，找到它的根节点加一
    if(vis[x]) x = find(x)+1;
    //输出修改后的数字并且标记
    cout <<x <<" ";
    vis[x] = true;
    //判断数字前后是否连续
    if(vis[x-1] && x != 1) fat[x-1] = x;
    if(vis[x+1]) fat[x] = x + 1;
  }
  return 0;
}

最后附上蓝桥杯汇总链接：蓝桥杯C/C++A组省赛历年真题题解 

声明：图片均来源于蓝桥杯官网，以个人刷题整理为目的，如若侵权，请联系删除~