本题基于一种贪心的思路,即:能不改就不改,但是一旦改了以后,后面的都改;具体而言,就是从前往后扫,如果当前位置的数不与前面的数冲突的话,那么我们就不改动他。但是,一旦有一个数冲突了,根据题意我们肯定要换一个更大的数,那么此时整个序列肯定就比原序列大了,那么后面的数我们肯定都要换尽可能小的。
同时,还要考虑的一个问题是,如何保证后面选的数不与前面已经选过的数冲突。方法非常暴力:我们预处理出每个数的所有约数,一旦该数被选中,那么我们就把他的所有约数的倍数都删除掉,也就是不可选。为了高效的进行查找可行数与删除操作,我们可以用一个set来存放所有数字。
PS:虽然题目给出的数据范围是1e5,但是并不保证符合条件的答案的数一定只在这个范围内,经测试2e6是安全的。
#include<cstdio>
#include<set>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
const int N=2e6+5;
bool isnot[N],erase1[N];
set<int>s;
vector<int>G[N];
void getys(void)//预处理约数
{
for(int i=2;i<N;i++){
//if(isnot[i])continue;
if(!isnot[i]) {
for (int j = i; j < N; j += i) {
isnot[j] = true;
G[j].push_back(i);
}
}
s.insert(i);//初始时从2到N都认为是可以的
}
}
int main()
{
int n,i,j,k;
cin>>n;
getys();bool islarge=false;
for(i=1;i<=n;i++){
int a,b;
scanf("%d",&a);b=*s.begin();//因为要输出,所有先赋值为最小的数
if(!islarge) {//如果还没有更改,也就是序列还没有大于原序列
b = *s.lower_bound(a);//用set可以高效查找当前剩下数中符合要求的数
if(b!=a)islarge=true;//如果找到的第一个满足要求的数不是原序列中的数的话,
// 也就是说原序列中的数与前面的数是冲突的,那么就说明这个位置已经发生了更改
}
cout<<b<<' ';
for(int j:G[b]){//删除选中数的所有约数的倍数
if(erase1[j])continue;
for(k=j;k<N;k+=j){
if(!erase1[k]){
s.erase(k);erase1[k]=true;
}
}
}
}
return 0;
}