Hossam and Trainees(分解质因数，欧拉筛)

最新推荐文章于 2024-04-03 23:55:07 发布

Cambrain_

最新推荐文章于 2024-04-03 23:55:07 发布

阅读量158

点赞数

分类专栏：数学 codeforces 文章标签：算法

本文链接：https://blog.csdn.net/m0_63761643/article/details/128569079

版权

数学同时被 2 个专栏收录

23 篇文章 0 订阅

订阅专栏

codeforces

16 篇文章 0 订阅

订阅专栏

题目链接：

Problem - 1771C - Codeforces

题目大意：

给定n个数，问是否存在不互质的两个数

思路：

很容易想到算数基本定理，将每个数分解质因子，查询是否存在一个质因子出现过两次即可。

复杂度 $O(n*\sqrt{A})$ 会TLE

可以优化每次分解质因数。先处理出 $\sqrt{A}$ 以内的质数。然后枚举质数即可。

复杂度 $O(n*\frac{\sqrt{A}}{logA})$

Code:

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<unordered_map>
using namespace std;
#define x first
#define y second
#define PII pair<int,int>
#define V vector<int>
#define endl "\n"
typedef long long ll;
typedef unsigned long long ull;
const int INF=0x3f3f3f3f;
const int N=1e5+10,M=4e4;
int prime[N],cnt;
bool st[N];
void sieve()
{
    for(int i=2;i<=M;i++)
    {
        if(!st[i]) prime[cnt++]=i;
        for(int j=0;prime[j]<=M/i;j++)
        {
            st[prime[j]*i]=true;
            if(i%prime[j]==0) break;
        }   
    }
}
void solve()
{
    unordered_map<int,int> mp;
    int n;
    cin>>n;
    int f=0;
    for(int i=1;i<=n;i++)
    {
        int x;
        cin>>x;
        for(int j=0;prime[j]<=x/prime[j];j++)
        {
            int p=prime[j];
            if(x%p==0)
            {
                if(mp[p]) f=1;
                mp[p]=1;
                while(x%p==0)
                {
                    x/=p;
                }
            }
        }
        if(x==1) continue;
        if(mp[x]) f=1;
        mp[x]=1;
    }
    if(f) cout<<"YES\n";
    else cout<<"NO\n";
    
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    sieve();
    int _t;
    cin>>_t;
    while(_t--) solve();
    return 0;
}