Power Products

链接：https://codeforces.com/contest/1247/problem/D

题意：

给定n个数组，a[i],求其中有多少个二元组(i,j),i!=j，a[i]*a[j]=x^k,x可以为任意数

解析：

如果a[i]*a[j]==x^k,呢么a[i]*a[j]的素因子和x的素因子相同，且a[i]*a[j]的素因子的幂次全部为k的倍数

我们把a[i]素因子分解，如果一个素因子的次方>k,我们直接取余，a[i]=pri[1]^c[1]*pri[2]^c[2]*....*pri[n]*c[n]

然后我们会得到一个数S，值为  S=pri[1]^(c[1]%k)*pri[2]*(c[2]%k)*....*pri[n]*(c[n]%k);它一定<a[i]

呢么和他相乘能为x^k的数G，G=pri[1]^(k-c[1]%k)*pri[2]*(k-c[2]%k)*....*pri[n]*(k-c[n]%k);

答案加上G已经被标记的数目，G可能非常大，我们在求的时候，一下一下乘，如果>1e5，就没有二元组

注意：算G乘的时候可能大于int

最后S标记++

ac：

#include<bits/stdc++.h>
#define ll long long
#define MAXN 200005
using namespace std;
int vis[MAXN]={0};
ll ans=0;

ll qpow(ll a,ll b)
{
    ll ans=1;
    while(b)
    {
        if(b&1) ans=(ans*a);
        a=(a*a);
        b>>=1;
    }
    return ans;
}

void get(int x,int k)
{
    int b[101],num=0;//素因子
    int c[101],cnt=0;//素的幂次
    int tmp=x;
    int ck=1;
    for(int i=2;i*i<=x;i++)//预处理1000内的素数表会更快
    {
        int g=0;
        while(tmp%i==0){
            g++;
            tmp=tmp/i;
        }
        g=g%k;
        if(g!=0){
            b[++num]=i;
            c[num]=g;
        }
        ck=ck*qpow(i,g);
    }
    if(tmp!=1){//a[i]本身为素数
        b[++num]=tmp;
        c[num]=1;
    }
    ll gg=1,sign=0;//gg一定要为ll
    for(int i=1;i<=num;i++)
    {
        int kk=b[i];
        for(int j=1;j<=k-c[i];j++){
            gg=gg*kk;
            if(gg>1e5){
                sign=1;
                break;
            }
        }
    }
    if(sign==0){
        ans+=vis[gg];
    }
    ck=ck*tmp;
    vis[ck]++;
}

int main()
{
    int n,x,k;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        get(x,k);
    }
    printf("%lld\n",ans);
    return 0;
}