暑假集训Day12 H (数论 STL)

题目链接在这里：Problem - H - Codeforces

这题思路不难想，将每个数分解质因数之后，如果有两个数，对应质因数的次数加起来正好是k的倍数，那这两个就是符合的，也就是说这两个数要形成一个互补的关系，用数组来实现的话会比较麻烦，使用map映射函数配合vector容器会简化很多

不得不说映射函数map是个非常牛批的东西，连vector都能映射！

#include "bits/stdc++.h"
using namespace std;
typedef long long LL;
const int MAX=1e5+5;
LL n,k,a[MAX],ans;
vector <pair<LL,LL> >x,y;
map<vector<pair<LL,LL> >,LL> b;
int main(){
    freopen ("h.in","r",stdin);
    freopen ("h.out","w",stdout);
    LL i,j,zt,cnt;
    scanf("%lld%lld",&n,&k);
    for (i=1;i<=n;i++) scanf("%lld",a+i);
    for (i=1;i<=n;i++){
        zt=a[i];
        x.clear(),y.clear();
        for (j=2;j*j<=zt;j++){
            cnt=0;
            while (zt%j==0){
                cnt++;
                zt/=j;
            }
            cnt%=k;
            if (cnt==0) continue;
            x.push_back(make_pair(j,cnt));
            y.push_back(make_pair(j,k-cnt));
        }
        if (zt!=1){
            x.push_back(make_pair(zt,1));
            y.push_back(make_pair(zt,k-1));
        }
        ans+=b[y];
        b[x]++;
    }
    printf("%lld",ans);
    return 0;
}