Codeforces Round #596 (Div. 2)D.Power Products

题意：

　　给一个数组，给你一个k，找出两个数字的积可以变成x^k的数对对数

解析：

　　当且仅当，两个数进行质因子分解后每个因子的个数都是k的倍数个就说明这是满足条件的一对，可以让每个因子个数%k用map找对应的数。

代码：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <vector>
#include <cstdio>
#include <queue>
#include <cmath>
#include <map>

using namespace std;

typedef long long LL;

const int mod=1e9+;
const int maxn=1e5+;

int n,k;

struct node{
    int x,num;
    node(){}
    node(int _x,int _num):x(_x),num(_num){}
    bool operator<(const node &s)const{
        if(x!=s.x)return x<s.x;
        else return num<s.num;
    }
};

vector<node>nodes;
map<vector<node>,int>mp;

void f(int x){
    int o=;
    for(int i=;i<=x/i;i++){
        if(x%i==){
            int num=;
            while(x%i==){
                num++;
                x/=i;
            }
            if(num%k!=){
                o=;
                nodes.push_back(node(i,num%k));
            }
        }
    }
    if(x!=)nodes.push_back(node(x,));
    if((int)nodes.size()==&&(o==||x==)){
        //nodes.push_back(node(x,1));
    }
}

int main(){
    scanf("%d%d",&n,&k);
    long long num=;
    for(int i=;i<n;i++){
        int x;
        scanf("%d",&x);
        nodes.clear();
        f(x);
        for(int i=;i<(int)nodes.size();i++){
            nodes[i].num=k-nodes[i].num;
        }
        if(mp[nodes]!=){
            num+=mp[nodes];
        }
        for(int i=;i<(int)nodes.size();i++){
            nodes[i].num=k-nodes[i].num;
            //printf("%d %d\n",nodes[i].x,nodes[i].num);
        }
        mp[nodes]++;
        //printf("%d     %lld\n",i,num);
    }
    printf("%lld\n",num);
    return ;
}