山东省第六届省赛 - Square Number（平方数）

题目

题意：

给你一些数，让你找到相乘是平方数的对数。

POINT：

先打个质数筛，不然超时。

因为平方数=平方数*平方数。

一开始我对每一个数枚举出它的平方数因子，找出对应的个数。这样超时。O(n*sqrt(Max)).

看了题解：

对每个数进行降平方操作，把平方数因子全部除掉。

最后得到num数组，C(num[x],2) (x=1,2,3……)就行了。

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <cmath>
using namespace std;
const int maxn = 100000+55;
#define LL long long
const LL mod = 1e9+7;

int num[1000100];
int a[1000010];
int is[1000100];
int pri[maxn];
int cnt=0;


void init()
{
	for(int i=2;i<=1000000;i++){
		if(is[i]==0){
			pri[++cnt]=i;
			for(int j=2*i;j<=1000000;j+=i)
				is[j]=1;
		}
	}
}


int f(int x)
{
	for(int i=1;pri[i]*pri[i]<=x;i++){
		while(x%(pri[i]*pri[i])==0)
			x/=pri[i]*pri[i];
	}
	return x;
}


int main()
{
	init();
	int T;
	scanf("%d",&T);
	while(T--){
		int n;
		scanf("%d",&n);
		memset(num,0,sizeof num);
		int ans=0;
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
			a[i]=f(a[i]);
			ans+=num[a[i]];
			num[a[i]]++;
		}
		printf("%d\n",ans);
	}





	return 0;
}