CF1497E1 Square-free division (easy version)(唯一分解定理+数论+思维)

题目传送门

题意

现在给你一串数字，请你划分最少的区间，使得这区间里面的所有数字两两相乘都不是平方数，请输出区间数。

思路

这道题可以用一个map来存储（当然要进行质因数的分解，要不然会卡常（唯一分解定理））因数为奇数的次数，因为偶数一定可以被开方。
这道题的实现方式我觉得还是比较精巧的

#include<iostream>
#include<map>
using namespace std;
const int N=1e7+9;
int primes[N], cnt;     // primes[]存储所有素数
bool st[N];         // st[x]存储x是否被筛掉
 
void get_primes(int n)
{
	n=N;
    for (int i = 2; i <= n; i ++ )
    {
        if (!st[i]) primes[cnt ++ ] = i;
        for (int j = 0; primes[j] <= n / i; j ++ )
        {
            st[primes[j] * i] = true;
            if (i % primes[j] == 0) break;
        }
    }
}
int bym(int n){
	int res=1;
	for(int i=0;primes[i]<=n/primes[i];i++){
		int u=0;
		while(n%primes[i]==0){
			n/=primes[i];
			u++;
		}
		if(u%2)res*=primes[i];
	}
	if(n!=1)res*=n;
	return res;
}
int main(){
	int T; 
	cin>>T;
	get_primes(N);
	while(T--){
		map<int,int>mp;
		int n,k;
		cin>>n>>k;
		int ans=1;
		for(int i=1;i<=n;i++){
			int s;
			scanf("%d",&s);
			int t;
				t=bym(s);
			if(mp[t]){
				mp.clear();
				ans++;
			//	mp[t]++;
			}
			mp[t]++;
		}
		cout<<ans<<endl; 
		//mp.clear();
	}
}