Codeforces Round #473 (Div. 2) D. Mahmoud and Ehab and another array construction task（构造）

题目链接：http://codeforces.com/contest/959/problem/D

题目大意：题目给出一个长度为n的数组a，要你构造出满足如下三个条件的长度也为n的数组b：

1、数组b的字典序大于等于a；

2、数组b中的任意元素>=2；

3、数组b中的任意两个元素都互质，也就是对于任意1<=i,j<=n,有gcd(b[i],b[j])=1;

同时题目要求b的字典序尽可能的小。

题目思路：要使任意两个元素的gcd等于1，那么就是使整个数组当中的任意一个数质因子都不相等，那么我们就可以用一个数组vis[i]来记录 i 这个因子是否是数组中某一个数的因子。

这样我们在构造的答案时候就可以先对a[i]进行质因子分解，判断a[i]的质因子是否在前i-1项里出现过

如果没出现，那么b[i]=a[i]，再将a[i]的所有质因子打上标记

如果出现过，因为要使b的字典序最小，那么就从a[i]+1到第一大于a[i]的素数开始枚举，找到第一个满足条件的值

特殊的情况就是枚举到第 j 位时，如果有一个b[i]>a[i]（1<=i<j）的话，那么就可以从2开始找第一个没出现过的素数填入b[j]即可（这是因为有b[i]>a[i]的条件时就已经满足b的字典序大于a了，那么要使b的字典序最小，只需要一位一位填入最小的素数即可）。

具体实现看代码吧

#include <bits/stdc++.h>
using namespace std;
const int MX=1e5+7;

int n;
int a[MX],b[MX];
vector<int>p;
int prime[10*MX],pcnt;
bool vis[20*MX];
void init() {
    for(int i=2; i<20*MX; i++) {
        if(!vis[i]) {
            prime[++pcnt]=i;
            for(int j=2*i; j<20*MX; j+=i)
                vis[j]=1;
        }
    }
}
void solve(int x,vector<int>&v) {
    for(int i=1; i<=pcnt&&x/prime[i]>=prime[i]; i++) {
        if(x%prime[i]==0) {
            v.push_back(prime[i]);
            while(x%prime[i]==0)
                x/=prime[i];
        }
    }
    if(x!=1)
        v.push_back(x);
}
bool check(int x) {
    p.clear();
    solve(x,p);
    for(int j:p)
        if(vis[j])
            return 0;
    return 1;
}
int main() {
    init();
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
        scanf("%d",&a[i]);
    memset(vis,0,sizeof(vis));
    int flag=0,las=1;
    for(int i=1; i<=n; i++) {
        if(!flag) {
            p.clear();
            solve(a[i],p);
            for(int j:p) {
                if(vis[j]) {
                    int pos=lower_bound(prime+1,prime+pcnt+1,a[i])-prime;
                    while(prime[pos]<=a[i])
                        pos++;
                    while(vis[prime[pos]])
                        pos++;
                    int tmp;
                    for(int k=a[i]+1; k<=prime[pos]; k++) {
                        if(check(k)) {
                            tmp=k;
                            break;
                        }
                    }
                    b[i]=tmp;
                    vector<int>v;
                    solve(tmp,v);
                    for(int k:v) vis[k]=1;
                    flag=1;
                    break;
                }
            }
            if(!flag) {
                for(int j:p)
                    vis[j]=1;
                b[i]=a[i];
            }
        } else {
            while(vis[prime[las]]&&las<=pcnt)
                las++;
            b[i]=prime[las];
            vis[prime[las]]=1;
        }
    }
    for(int i=1; i<=n; i++)
        printf("%d%c",b[i],i==n?'\n':' ');
    return 0;
}