Codeforces Round #708 (Div. 2) E1-Square-free division (easy version)(素数筛+思维)

题意:

给出一个数组,将这个数组分成最少的连续段,使得每段中的数字两两相乘不为平方数。

题解:

因为所有数都可以表示成素数相乘,所以先用欧拉筛预处理素数,然后对每个数都质因子分解。

那么 假设 X = a x ⋅ b y ⋅ c z . . . . X=a^x \cdot b^y \cdot c^z .... X=axbycz.... ,假设 x 和 z x和z xz 为奇数,既然不能成为平方数,也就是说,不能出现一个数,使得与 X X X 相乘后, a a a c c c 的指数变为偶数。那么这个数的要求是什么?显然,质因子分解后,奇数次幂的数不能恰好为 a a a c c c

那么我们用 m a p map map存一下, m a p [ a ⋅ c ] = 1 map[a \cdot c]=1 map[ac]=1 ,即表示这个数后面的数,奇数次幂的质数相乘不能再map中出现。满足要求就放在同一段,否则就放在新的段,并且清空 m a p map map

代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int>pii;
const int MAXN=1e7;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
int prime[MAXN],cnt;
int isprime[MAXN];
int a[200005];
map<int ,int >mp;
void getprime(int n)
{
    
    
    for(ll i=2;i<=n;i++)
    {
    
    
        if(!isprime[i])
        {
    
    
            prime[++cnt]=i;
            isprime[i]=i;
        }
        for(int j=1;j<=cnt&&i*prime[j]<=n;j++)
        {
    
    
            isprime[i*prime[j]]=prime[j];
            if(i%prime[j]==0) break;
        }
    }
}
int solve(int temp)
{
    
    
    int pre=isprime[temp];
    int res=1,cnt=0;
    while(temp!=1)
    {
    
    
        int now=isprime[temp];
        if(now!=pre)
        {
    
    
            if(cnt&1) res=res*pre;
            cnt=0;
        } 
        cnt++;
        temp/=now;
        pre=now;
    }
    if(cnt&1) res=res*pre;
    return res;
}
int main()
{
    
    
    getprime(1e7);
    int t;
    scanf("%d",&t);
    while(t--){
    
    
        int n,k;
        scanf("%d%d",&n,&k);
        int ans=0;
        for(int i=1;i<=n;i++){
    
    
            scanf("%d",&a[i]);
        }
        for(int i=1;i<=n;i++){
    
    
            int temp=a[i];
            int res=solve(temp);
            if(mp.count(res))
            {
    
    
                ans++;
                mp.clear();
                mp[res]=1;
                
            }
            else
            {
    
    
                mp[res]=1;
            }
        }
        printf("%d\n",ans+1);
        mp.clear();
    }
}

/*
3
5 0
18 6 2 4 1
*/

猜你喜欢

转载自blog.csdn.net/weixin_45755679/article/details/114962027