题解 - CF GCD Groups 2

题解 - CF GCD Groups 2

题目链接： Codeforces Round #576 (Div. 1)

题意：

给你一个长度为N的序列，要求将其分为两组，每组的元素不少于两个，并使得两组的GCD都为1，求出一种分配方案。

数据范围： $2\le n\le 10^5$ $1\le a_i\le 10^9$

---

思路

因为两边都要求GCD为一，说明每组都没有相同的质因子，而一个 $(1,10^9)$ 内的数最多只会有9个质因子，所以质因子的分布是十分稀疏的。如果存在一种合理的分配方案，那么可以先随机一个数放入集合一，如果每次加入的一个数可以将其GCD减少，那么就将其加入，否则就将其放入集合二，而集合二大概率也是gcd为一的，因为集合里面的数字越多，集合二gcd为一的可能性也越大，而集合一的gcd已经尽可能小了，所以这样分配的正确性是合理的，我们就多随机几次，洗的越均匀越好。

---

代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+7;
int n,s;
pair<int,int> a[N];
int belong[N];
int main(){
    int start=clock();
    scanf("%d",&n);
    for(int i=1;i<=n;++i){
        scanf("%d",&a[i].first);
        a[i].second=i;
        s=__gcd(s,a[i].first);
    }
    if(s>1){
        puts("NO");
        return 0;
    }
    mt19937 rnd(time(0));//随机数
    while(1.0*(clock()-start)/CLOCKS_PER_SEC<0.45){ //如果时限超过了0.45就停止
        memset(belong,0,sizeof(belong));
        shuffle(a+1,a+n+1,rnd);//随机打乱
        int x=a[1].first,y=0;
        belong[a[1].second]=1;
        for(int i=2;i<=n;++i){
            if(a[i].first%x==0){
                y=__gcd(y,a[i].first);
                belong[a[i].second]=2;
            }else{
                x=__gcd(x,a[i].first);
                belong[a[i].second]=1;
            }
        }
        if(x==1&&y==1){
            puts("YES");
            for(int i=1;i<=n;++i){
                printf("%d ",belong[i]);
            }
            puts("");
            return 0;
        }
    }
    puts("NO");
    return 0;
}