数论+思维

给定一个长度为 nn 的正整数数列 a1,a2,…,ana1,a2,…,an 和一个正整数 kk。

请你判断共有多少个数对 (l,r)(l,r) 同时满足：

• 1≤l<r≤n1≤l<r≤n
• 存在一个整数 xx 使得 al×ar=xkal×ar=xk 成立

输入格式

第一行包含两个整数 n,kn,k。

第二行包含 nn 个正整数 a1,a2,…,ana1,a2,…,an。

输出格式

一个整数，表示满足条件的数对的数量。

数据范围

前三个测试点满足 2≤n≤102≤n≤10。
所有测试点满足 2≤n≤1052≤n≤105，2≤k≤1002≤k≤100，1≤ai≤1051≤ai≤105。

//质因数分解，指数r mod k 后找k-r的指数
//分解可以只分解根号n 因为只有一个质因数大于根号n特判一下即可
#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
typedef long long ll;
const int maxm=1e3+5;
const int maxn=1e5+5;
int a[maxn];int b[maxn];
int prim[maxn];
int pf[maxn];
int cnt=0;
void ol()
{
    for(int i=2;i<maxm;i++)
    {
        //cout<<i<<endl;
        if(!pf[i]) prim[++cnt]=i;
        for(int j=1;j<=cnt&&prim[j]*i<maxm;j++)
        {
            pf[prim[j]*i]=1;
            if(i%prim[j]==0) break;
        }
    }
}
ll qpow(ll a,ll b)
{
    int res=1;
    while(b)
    {
        if(a>maxn) return 0;
        if(b&1)
        {
            res*=a;
            if(res>maxn) return 0;
        }
        b>>=1;
        a=a*a;
    }
    return res;
}
int main()
{
    int n,k;scanf("%d%d",&n,&k);
    ol();
    //cout<<cnt<<endl;
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    ll res=0;
    for(int i=0;i<n;i++){
        int t=a[i];
        ll now=1;
        ll pre=1;
        for(int j=1;prim[j]*prim[j]<=a[i];j++)
        {
            int num=0;
            while(t%prim[j]==0)
            {
                //cout<<prim[j]<<endl;
                t/=prim[j];
                num++;
            }
            num=num%k;
            if(num==0) continue;
            now*=qpow(prim[j],num);
            pre*=qpow(prim[j],k-num);
            if(pre>maxn) pre=0;
            if(t==1) break;
        }
        if(t!=1) now=now*t,pre*=qpow(t,k-1);
        if(pre>maxn) pre=0;
        res+=b[pre];
        b[now]++;
    }
    printf("%lld\n",res);
}