Codeforces_959D_Mahmoud and Ehab and another array construction task（數學+貪心）

傳送門

題意：給你一個序列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;
}