1. 程式人生 > >Codeforces_959D_Mahmoud and Ehab and another array construction task(數學+貪心)

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; }