CF1771C

Problem - C - Codeforces

大意：只要一组数据中存在两个数有公约数即输出YES，否则输出NO。

思路：

由素数定理可得：任何一个自然数都可以拆成若干个质数的积。

而每一组最多只有1e5个数，所有由抽屉原理

我们可以先进行预处理筛出 1e5以内的素数，然后暴力筛即可

代码：

#include <bits/stdc++.h>
#include <unordered_map>
#include <unordered_set>

using namespace std;
#define pi acos(-1.0)
#define endl '\n'
#define x first
#define y second
typedef long long LL;
priority_queue<int> pq;
priority_queue<int, vector<int>, greater<int> > up;
deque<int> dq;
typedef pair<LL, LL> PII;
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
const int inf = 0x3f3f3f3f;
const int N = 1e5 + 10, M = 1e9;

int cnt, primes[N];
bool st[N];

void get_primes(int n)  // 线性筛法
{
    for(int i = 2; i <= n; i ++)
    {
        if(!st[i]) primes[cnt ++] = i;
        // 将已经筛出来的质数的倍数筛掉
        for(int j = 0; primes[j] * i <= n; j ++)
        {
            st[primes[j] * i] = true;
            if(i % primes[j] == 0) break;
        }
    }
}

void solve()
{
    int n;
    cin >> n;
    vector<int> a(n);
    for (int i = 0; i < n; i ++ ) cin >> a[i];
    
    set<int> s;
    for (int i = 0; i < n; i ++ )
    {
        int x = a[i];
        for(auto tt : primes){
            if(tt * tt > x) break;
            
            if(x % tt == 0) {
                if(s.contains(tt))  // 判断集合中是否已经存在这个因子
                {
                    cout << "YES" << endl; return;
                }
                s.insert(tt);
                while(x % tt == 0) x /= tt;
            }
        }
        if(x > 1) {
            if(s.contains(x)) {
                cout << "YES" << endl; return;
            }
            s.insert(x);
        }
    }
    cout << "NO" << endl;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(0);
    
    get_primes(100000);
    int T = 1;
    cin >> T;
    while(T --)
    {
        solve();
    }
    
    return 0;
}