題目鏈接:https://codeforces.com/contest/1247/problem/D
題目大意:給n個數字,問有多少對數字的乘積是某個數字的k次方
題目思路:很明顯,一個數字的k次方需要滿足的條件是他的每個質因數的冪次都是k的倍數,那麼只要得出當前質因數的冪次情況,看看能把所有質數的冪次都補成k的倍數的數字個數就行
這裏唯一想不到的點就是map居然能套一個vector,vector裏面再能套一個pair,實在太神奇了!
以下是代碼:
#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
const int MAXN = 2e5+5;
const int MOD = 1e9+7;
int n,k;
int x;
map<vector<pair<int,int> >,int>mp;
vector<pair<int,int> >v;
pair<int,int>p;
int main(){
while(cin>>n>>k){
mp.clear();
ll ans=0;
rep(i,1,n){
v.clear();
cin>>x;
rep(i,2,sqrt(x)){
if(x%i==0){
int num=0;
while(x%i==0)num++,x/=i;
num%=k;
if(num){
p.first=i,p.second=num;
v.push_back(p);
}
}
}
if(x!=1){
p.first=x,p.second=1;
v.push_back(p);
}
ans+=mp[v];
int len=v.size();
rep(i,0,len-1){
v[i].second=k-v[i].second;
}
mp[v]++;
}
cout<<ans<<endl;
}
return 0;
}