题意:
给你一些数,让你找到相乘是平方数的对数。
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;
}