蓝桥杯-修改数组

题目描述

修改数组

---

题解：

利用常规方法遍历复杂度过大，通过分析题目依次累加1可以想到利用并查集的树形结构进行查找赋值。
并查集的构建-详解并查集
首先初始化10的六次方个节点，接着开始输入，每次查找输入元素的根并将根的值赋值给a[i]，接着更新a[i]的根+1：再次访问时根植大1，这样就会不断向上爬，利用树形结构直接查找并赋值。
代码如下:

#include<iostream>
using namespace std;
const int maxx=1000010;
int par[maxx];
int ran[maxx];
int a[maxx];
void init(int n)
{
	for (int i = 1; i <= n; i++)
	{
		par[i]=i;
		ran[i]=0;
	}
}
int find(int x)//路径压缩
{
	if (par[x]==x)
	{
		return x;
	}
	else
	{
		return par[x]=find(par[x]);
	}
}
int find2(int x)//单纯查找
{
	while (par[x]!=x)
	{
		x=par[x];
	}
	return x;
}
void unite0(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x!=y) par[x]=y;
}
void unite(int x,int y)
{
	x=find(x);
	y=find(y);
	if (x==y)
	{
		return ;
	}
	if (ran[x]<ran[y])
	{
		par[x]=y;
	}
	else
	{
		par[y]=x;
		if(ran[x]==ran[y]) ran[x]++;
	}
}
bool same(int x,int y)
{
	return find(x)==find(y);
}
int main()
{
	init(1000000);
	int n,k;
	cin>>n;
	for (int i = 0; i < n; i++)
	{
		cin>>k;
		a[i]=find(k);
		par[a[i]]=a[i]+1;
	}
	for (int i = 0; i < n; i++)
	{
		cout<<a[i]<<' ';
	}
	return 0;
}

---