山东省第六届ACM省赛 H---Square Number 【思考】

#include <stdio.h>
#include  <string.h>

bool prime(int x)
{
    for(int i=2; i*i<=x; i++){
        if(x%i==0) return false;
    }
    return true; //判断素数
}

int a[1000], cnt[1000000+10];

int main()
{
    //将100万以内的"素数的完全平方数"打表
    int e=0;
    for(int i=2; i*i<=1000000; i++){
        if(prime(i))
            a[e++]=i*i;
    }
    int tg; scanf("%d", &tg);
    int n, i, j;
    while(tg--)
    {
        scanf("%d", &n);
        int dd;
        int mm=1;
        memset(cnt,0,sizeof(cnt));
        while(n--){
            scanf("%d", &dd);
            for(i=0; i<e&&a[i]<=dd; i++){
                while(dd%a[i]==0)
                    dd/=a[i];
            }
            if(dd>mm) mm=dd;
            cnt[dd]++;
        }
        long long ans=0;
        for(j=1; j<=mm; j++){
            ans+=(cnt[j]*(cnt[j]-1)/2);
        }
        printf("%lld\n", ans);
    }
}