P8686

修改数组

题目描述

给定一个长度为 \(N\) 的数组 \(A=[A_1,A_2, \cdots A_N]\),数组中有可能有重复出现的整数。

现在小明要按以下方法将其修改为没有重复整数的数组。小明会依次修改 \(A_2,A_3, \cdots ,A_N\)

当修改 \(A_i\) 时,小明会检查 \(A_i\) 是否在 \(A_1\)\(A_{i-1}\) 中出现过。如果出现过,则小明会给 \(A_i\) 加上 \(1\);如果新的 \(A_i\) 仍在之前出现过,小明会持续给 \(A_i\)\(1\),直到 \(A_i\) 没有在 \(A_1\)\(A_{i-1}\) 中出现过。

\(A_N\) 也经过上述修改之后,显然 \(A\) 数组中就没有重复的整数了。

现在给定初始的 \(A\) 数组,请你计算出最终的 \(A\) 数组。

输入格式

第一行包含一个整数 \(N\)

第二行包含 \(N\) 个整数 \(A_1,A_2, \cdots ,A_N\)

输出格式

输出 \(N\) 个整数,依次是最终的 \(A_1,A_2, \cdots ,A_N\)

样例 #1

样例输入 #1

5
2 1 1 3 4

样例输出 #1

2 1 3 4 5

提示

对于 \(80\%\) 的评测用例,\(1 \le N \le 10000\)

对于所有评测用例,\(1 \le N \le 10^5\)\(1 \le A_i \le 10^6\)

这题用并查集做就……
思路
若fa[a[i]]=a[i]说明没有出现过就是本身
否则直接在集合内找到root即可
相当于所有相邻的构成一个联通块 块的root就是maxx+1
查询每个数在哪个联通块即可
点击查看代码
#include<bits/stdc++.h>
using namespace std;
int n,a[100005],b[100005],m,maxx;
int fa[100005];
int getfa(int x)
{
	if(x==fa[x])return x;
	return fa[x]=getfa(fa[x]);
}
void uni(int x,int y)
{
	int fx=getfa(x),fy=getfa(y);
	if(x<y)fa[x]=y;
	if(y<x)fa[y]=x;
}
int main()
{
	ios::sync_with_stdio(false);
	cin>>n;
	for(int i=1;i<=n;i++)cin>>a[i];
	for(int i=1;i<=100005;i++)fa[i]=i; 
	for(int i=1;i<=n;i++)
	{
		b[i]=getfa(a[i]);
		uni(b[i],b[i]+1);
	}
	for(int i=1;i<=n;i++)cout<<b[i]<<" ";
	return 0;
}
posted @ 2023-01-19 18:20  PKU_IMCOMING  阅读(36)  评论(0)    收藏  举报