Codeforces_959D_Mahmoud and Ehab and another array construction task(数学+贪心)

版权声明: https://blog.csdn.net/yyy_3y/article/details/79965564

传送门

题意:给你一个序列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;
}

猜你喜欢

转载自blog.csdn.net/yyy_3y/article/details/79965564