蓝桥杯 修改数组【第十届】【省赛】【A组】C++

资源限制

内存限制：256.0MB C/C++时间限制：1.0s Java时间限制：3.0s Python时间限制：5.0s

问题描述

　　给定一个长度为 N 的数组 A = [A₁, A₂, · · · AN]，数组中有可能有重复出现的整数。
　　现在小明要按以下方法将其修改为没有重复整数的数组。小明会依次修改
　　A₂, A₃, · · · , AN。
　　当修改 Ai 时，小明会检查 Ai 是否在 A₁ ∼ Ai−₁ 中出现过。如果出现过，则小明会给 Ai 加上 1 ；如果新的 Ai 仍在之前出现过，小明会持续给 Ai 加 1 ，直到 Ai 没有在
　　A₁ ∼ Ai−₁ 中出现过。
　　当 AN 也经过上述修改之后，显然 A 数组中就没有重复的整数了。现在给定初始的 A 数组，请你计算出最终的 A 数组。

输入格式

　　第一行包含一个整数 N。
　　第二行包含 N 个整数 A₁, A₂, · · · , AN 。

输出格式

　　输出 N 个整数，依次是最终的 A₁, A₂, · · · , AN。

样例输入

5
2 1 1 3 4

样例输出

2 1 3 4 5

评测用例规模与约定

　　对于 80% 的评测用例，1 ≤ N ≤ 10000。
　　对于所有评测用例，1 ≤ N ≤ 100000，1 ≤ Ai ≤ 1000000。

优化前70分Code

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int a[N];
int last[N];//last[i]必须>=i才行 
int n;
int main(){
scanf("%d",&n);
fill(last+1,last+n+1,n+1);//初始化，表示数字i暂时未出现，保证第一次出现last[a[i]>i
for(int i=1;i<=n;i++){
	scanf("%d",&a[i]);
	while(last[a[i]]<i)++a[i];
	last[a[i]]=i;//更新a[i]的数字最后一次出现的位置 
}

for(int i=1;i<=n;i++)
printf("%d ",a[i]);
return 0;
}

优化后AC Code

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int a[N];//a[i]-1表示数字i需要+a[i]-1才能出现左边未出现的数字  
int last[N]; //更新完后的数组 
int n;
int main(){
scanf("%d",&n);
int t;
for(int i=1;i<=n;i++){
	scanf("%d",&t);//暂存 
	while(a[t]){//判断t未出现过 
		a[t]++;//出现过就++
		t+=a[t]-1; //更新t 
	}
	a[t]++;//更新完t,a[t]=0,++标记t第一次出现。 
	last[i]=t; 
	
}
for(int i=1;i<=n;i++)
printf("%d ",last[i]);
return 0;
}