Codeforces Round #708 (Div. 2) E1. Square-free division (easy version)

传送门

题目大意

将一个有n个正整数的数组切分为数个数组，要求新的数组内元素两两相乘不能是完全平方数

题解

将n个数进行质因数分解，如果质因数的数量为偶数个，舍弃，如果质因数数量为奇数个，那么就保存一个，例如500，有2个2，3个5，那么我们将其变为5，这样从前往后遍历数组，如果每个段中有相同元素，分割成新段就好了，因为相同的数会组合出完全平方数，注意不能直接找出现次数最多的数当答案，因为可能有5 5 5 2 2 2 5这种情况

#include<cstdio>
#include<map>
using namespace std;
const int MAX=2e5+5;
const int MAXP=1e7;
int findprime[MAXP];
int prime[MAXP];
int primen;
int init(){
	for(int i=2;i<MAXP;i++){
		if(!findprime[i]){
			prime[primen++]=i;
			for(int j=i;j<MAXP;j+=i){
				findprime[j]=i;
			}
		}
	}
}
int main(){
	int t,n,k,m,pn,last,x;
	int num[MAX];
	init();
	scanf("%d",&t);
	while(t--){
		scanf("%d%d",&n,&k);
		for(int i=0;i<n;i++){
			m=1;
			last=0;
			pn=0;
			scanf("%d",&x);
			while(x>1){
				if(last==findprime[x]){
					pn++;
				}else{
					m*=pn%2?last:1;
					pn=1;
					last=findprime[x];
				}
				x/=findprime[x];
			}
			num[i]=m*(pn%2?last:1);
		}
		map<int,int> mlast;
		int ans=1,end=0;
		for(int i=0;i<n;i++){
			if(mlast.find(num[i])!=mlast.end()&&mlast[num[i]]>=end){
				ans++;
				end=i;
			}
			mlast[num[i]]=i;
		}
		printf("%d\n",ans);
	}
}