【蓝桥杯】2019年第十届真题 修改数组

题目 2301: [蓝桥杯][2019年第十届真题]修改数组

时间限制: 1Sec 内存限制: 128MB 提交: 2908 解决: 709

题目描述

给定一个长度为 N 的数组 A = [A1, A2, · · · AN ]，数组中有可能有重复出现 的整数。

现在小明要按以下方法将其修改为没有重复整数的数组。小明会依次修改 A2,A3,··· ,AN。

当修改 Ai 时，小明会检查 Ai 是否在 A1 ∼ Ai−1 中出现过。如果出现过，则 小明会给 Ai 加上 1 ;如果新的 Ai 仍在之前出现过，小明会持续给 Ai 加 1 ，直 到 Ai 没有在 A1 ∼ Ai−1 中出现过。

当 AN 也经过上述修改之后，显然 A 数组中就没有重复的整数了。 现在给定初始的 A 数组，请你计算出最终的 A 数组

输入

第一行包含一个整数 N。 第二行包含N个整数A1,A2,··· ,AN

 

对于 80% 的评测用例，1 ≤ N ≤ 10000。

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

 

输出

输出N个整数，依次是最终的A1,A2,··· ,AN。

样例输入

5
2 1 1 3 4

样例输出

2 1 3 4 5

题目解析

1. 时间超限

#include <iostream>
#include <unordered_set>
using namespace std;

int main(int argc, char** argv) {
	int n;
	cin >> n;
	int data[n];
	for(int i = 0; i < n; i++){
		cin >> data[i];
	}
	unordered_set<int> ust;
	for(int i = 0; i < n; i++){
		while(ust.count(data[i])){
			data[i]++;
		}
		ust.insert(data[i]);
	}
	for(int i = 0; i < n; i++){
		cout << data[i] << " ";
	}
	
	return 0;
}

2. 100分 并查集

visited[x] = k 记录访问x的次数， 那么它后面的k个数也一定被访问过了，所以直接对x + k，将x 调到x+k的位置在进行判断是否被访问了

#include <iostream>
using namespace std;

int A[100005];
int n;
int visited[1000055];

int main(int argc, char** argv) {
	for(int i = 0; i < 100005; i++){
		A[i] = 0;
	}
	for(int i = 0; i < 1000055; i++){
		visited[i] = 0;
	}
	
	int n;
	cin >> n;
	for(int i = 0; i <n; i++){
		int x;
		cin >> x;
		while(visited[x] != 0){
			visited[x] ++;
			x += (visited[x] - 1);
		}
		A[i] = x;
		visited[x]++;
	}
	for(int i = 0; i < n; i++){
		cout << A[i] << " "; 
	}

	return 0;
}