[Codeforces 1246B] Power Products (STL+分解质因数)

Posted 2020-11-16 birchtree

[Codeforces 1246B] Power Products (STL+分解质因数)

　题面

给出一个长度为(n)的序列(a_i)和常数k,求有多少个数对((i,j))满足(a_i imes a_j = x^k (x in mathbb{N}^+))。即这两个数乘起来恰好为一个正整数的(k)次方

(a_i,n leq 10^5)

分析

考虑(x^k)的质因数分解式 , 那么每一项的指数一定是k的倍数，即 (k|x_i).

因此对于每个 (a_i), 把它的质因数分解结果(sum p_i^{x_i})记录在一个数组中，每个元素形如((p_i, x_i mathrm{mod} k ))

那么 (a_j)的质因数分解为 $sum p_i^{k-x_i ?mathrm{mod} k } $.

所以只要求有多少个数组满足每一项都是$ (p_i,k-x_i mathrm{mod}?k) $.用 map< vector< pair<int,int> >, int> cnt; 存储就可以了

每个数组的大小为 (O(log n)) ,总时间复杂度为 (O(n log ^2 n))

　代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<map> 
#include<vector> 
#define maxv 100000
#define maxn 100000 
using namespace std;
typedef long long ll;
int n,k;
map< vector< pair<int,int> >, int> cnt; 
vector< pair<int,int> >p1,p2;
int a[maxn+5];
int vis[maxv+5];
int minprime[maxv+5];
int prime[maxv+5];
 
 
void sieve(int n) {
    vis[1]=1;
    for(int i=2; i<=n; i++) {
        if(!vis[i]) {
            minprime[i]=i;
            prime[++k]=i;
        }
        for(ll j=1; j<=k&&(ll)i*prime[j]<=n; j++) {
            minprime[(ll)i*prime[j]]=prime[j];
            vis[(ll)i*prime[j]]=1;
            if(!(i%prime[j])) break;
        }
    }
}
 
void divide(int x) {
    if(!vis[x]) {
        p1.push_back(make_pair(x,1));
        p2.push_back(make_pair(x,k-1)); 
        return;
    }
    while(x>1) {
        int t=minprime[x],u=0;;
        while(x%t==0&&x!=1) {
            x/=t;
            u=(u+1)%k;
        }
        if(u!=0){ 
            p1.push_back(make_pair(t,u));
            p2.push_back(make_pair(t,k-u));
        } 
    }
}
 
void print(vector< pair<int,int> > &x){
    for(int i=0;i<x.size();i++){
        printf("(%d,%d) ",x[i].first,x[i].second);
    }
    printf("
");
}
int main(){
    sieve(100000);
    scanf("%d %d",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    ll ans=0;
    for(int i=1;i<=n;i++){
        p1.clear();
        p2.clear();
        divide(a[i]);
//      print(p1);
//      print(p2);
        ans+=cnt[p2];
        cnt[p1]++;
        
    }
    printf("%I64d
",ans);
}