SDUT 3258 Square Number 简单数学

和上一题一样，把平方因子除去，然后对应的数就变成固定的

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long LL;
const int N=1e6+;
const int INF=0x3f3f3f3f;
int vis[N],prime[],cnt;
void getprime(){
  bool v[];
  memset(v,,sizeof(v));
  for(int i=;i*i<=;++i){
    if(v[i])continue;
    for(int j=i*i;j<=;j+=i)
      v[j]=;
  }
  for(int i=;i<=;++i)
    if(!v[i])prime[++cnt]=i;
}
int main(){
    int T,n;
    getprime();
    scanf("%d",&T);
    while(T--){
      memset(vis,,sizeof(vis));
      scanf("%d",&n);
      LL ans=;
      for(int i=;i<=n;++i){
        int t;scanf("%d",&t);
        for(int j=;j<=cnt&&prime[j]<=t/prime[j];++j){
            int tmp=;
            while(t%prime[j]==)t/=prime[j],++tmp;
            if(tmp&)t*=prime[j];
        }
        ans+=vis[t];
        // printf("%I64d\n",ans);
        ++vis[t];
      }
      printf("%lld\n",ans);
    }
    return ;
}