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

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

题面

给出一个长度为\(n\)的序列\(a_i\)和常数k,求有多少个数对\((i,j)\)满足\(a_i \times 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("\n");
}
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\n",ans);
}