Codeforces Round #596 (Div. 2, based on Technocup 2020 Elimination Round 2) D. Power Products(數論)

題目鏈接: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;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章