SDUT 3258 Square Number

来源：互联网发布：淘宝网天天特价女装编辑：程序博客网时间：2024/05/22 21:32

给出n,n≤1e6个数ai≤1e6，问其中有多少个数对满足这个数对的乘积是平方数

对于一个数x=∏i=1npeii

我们记sim(x)=∏i=1npeimod2i

那么其实一个数对(x,y)能满足他们的乘积是平方数当且仅当sim(x)=sim(y)

然后就可以用哈希了

代码如下

#include<bits/stdc++.h>using namespace std;const int maxn = 1123456;#define LL long longint pri[maxn];void sieve(){    memset(pri,-1,sizeof(pri));    pri[1] = 1;    for(LL i=2;i<maxn;i++){        if(pri[i] == -1){            pri[i] = i;            for(int j=i*2;j<maxn;j+=i){                if(pri[j] == -1) pri[j] = i;                else pri[j] *= i;            }            int cnt = 0;            for(LL j=i*i;j<maxn;j*=i){                cnt++;                for(LL k=1;k*j<maxn;k++){                    if(cnt%2) pri[k*j] /= i;                    else pri[k*j] *= i;                }            }        }    }}int Hash[maxn];int main(){    int T;    scanf("%d",&T);    int n;    sieve();    while(T-- && ~scanf("%d",&n)){        int x;        LL ans = 0;        memset(Hash,0,sizeof(Hash));        while(n--){            scanf("%d",&x);            x = pri[x];            ans += Hash[x];            Hash[x]++;        }        printf("%lld\n",ans);    }    return 0;}

0 0