“浪潮杯”山东省第六届ACM大学生程序设计竞赛 Square Number

题目：给你若干个数，两两相乘组成平方数的有多少个。

思路：一个合数肯定能拆成若干个素数相乘，我们只要除以质数的偶数次幂，找到能与他匹配的余数就可以了

代码：

#include<bits/stdc++.h>
using namespace std;
int dp[1005];
int pri[1005];
int num;
int k[1000005];
void prime()
{
    memset(pri,0,sizeof(pri));
    for(int i=2;i<=1000;i++)
    {
        for(int j=i+i;j<=1000;j+=i)
        {
            if(!pri[j])
            {
                pri[j]=1;
            }
        }
    }
    num=0;
    for(int i=2;i<=1000;i++)
    {
        if(!pri[i])
        {
            dp[num++]=i*i;
        }
    }
}
int main()
{
    prime();
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        memset(k,0,sizeof(k));
        int ans=0;
        for(int i=0;i<n;i++)
        {
            int x;
            scanf("%d",&x);
            for(int i=0;i<num;i++)
            {
                while(x%dp[i]==0)
                    x/=dp[i];
            }
            ans+=k[x];
            k[x]++;
        } printf("%d\n",ans);
    }

}