给定一个长度为 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);
}