Codeforces_959D_Mahmoud and Ehab and another array construction task(數學+貪心)
阿新 • • 發佈:2019-01-23
題意:給你一個序列A,讓你求一個序列B,滿足以下情況:
1.B的字典序比A大
2.Bi>=2
3.B中的沒兩個數都兩兩互質。
思路:先預處理質因數。B中滿足兩兩互質等價於B中沒有相同的質因數。所以對於每個數看,他的質因數有沒有在之前出現過,出現過就找 下一個,如果找到的數和起始的不一樣的話,之後的數只需要從最小的質數開始就行。
#include<bits/stdc++.h>
#define debug(a) cout << #a << " " << a << endl
#define LL long long
#define ull unsigned long long
#define PI acos(-1.0)
#define eps 1e-6
const int N=2e6+7;
using namespace std;
bool vis[N];
vector<int> stk[N];
void init()
{
for(int i=2;i<N;i++){
if(!vis[i]){
for(int j=i;j<N;j+=i) vis[j]=1,stk[j].push_back(i);
}
}
memset(vis,0,sizeof(vis));
}
int solve(int x)
{
for(auto &i:stk[x])
if(vis[i]) return 0;
for(auto &i:stk[x])
vis[i]=1;
return 1;
}
int main ()
{
//yyy_3y
//freopen("1.in","r",stdin);
int n;scanf("%d",&n);
init();
int flag=1;
int q=2;
for(int i=1;i<=n;i++){
int x; scanf("%d",&x);
int tmp=x;
if(flag){
while(!solve(tmp))tmp++;
if(tmp!=x) flag=0;
printf("%d ",tmp);
}else{
while(!solve(q)) q++;
printf("%d ",q);
}
}
return 0;
}