stl+数论——1247D

其实也不算很难想，每个元素质因子分解后的p^c的p和c用pair的形式存在每个元素vector里

要去前面找一个数使得所有指数相加是k的倍数，那么把vector里的所有c 模 k，然后去找前面互补的数的个数，可以用map存下前面元素模完k之后的vector，然后答案加一加就行

注意：如果a本身就是一个k次数，即其所有质因子的指数都是k的倍数，那么这种情况要单独处理一下

套了三个stl，一开始不敢写。。。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define N 100005
ll n,k,a[N],x,cnt[N];

int vis[N],prime[N],m;
void init(){
    for(int i=;i<=;i++){
        if(!vis[i]){
            prime[++m]=i;
        }
        for(int j=;j<=m;j++){
            if(prime[j]*i>)break;
            vis[prime[j]*i]=;
            if(i%prime[j]==)break;
        }
    }
}

ll p[N],c[N],mm;
void divide(ll x){
    mm=;
    for(int i=;i<=m;i++){
        if(prime[i]>x)break;
        if(x%prime[i]!=)continue;
        p[++mm]=prime[i];c[mm]=;
        while(x%prime[i]==)
            x/=prime[i],c[mm]++;
    }
    if(x>)
        c[++m]=,p[mm]=x;
}

map<vector<pair<int,int> >,int>mp;
vector<pair<int,int> >v[N];

int main(){
    init();

    cin>>n>>k;
    ll ans=,tot=;

    for(int i=;i<=n;i++){
        cin>>a[i];
        divide(a[i]);
        for(int j=;j<=mm;j++){
            c[j]%=k;
            if(c[j]!=)
                v[i].push_back(make_pair(p[j],c[j]));
        }

        if(v[i].size()==){
            ans+=tot;
            tot++;
            continue;
        }

        vector<pair<int,int> >tmp;
        for(int j=;j<v[i].size();j++){
            pair<int,int> p;
            p.first=v[i][j].first;
            p.second=k-v[i][j].second;
            tmp.push_back(p);
        }
        ans+=mp[tmp];

        mp[v[i]]++;
    }

    cout<<ans<<'\n';
}