HDU 5525 Product 数论

题意:

给出一个长度为(n(1 leq n leq 10^5))的序列(A_i)(N=prodlimits_{i=1}^{n}i^{A_i})。求(N)的所有约数的乘积。

分析:

首先还是将(N)质因数分解,(N=prodlimits_{i=1}^{k}p_i^{e_i})
考虑素因子(p_a)对结果的贡献,含有且仅含有(p_a^x)的约数有(frac{prodlimits_{i=1}^{k}e_i+1}{e_a+1})个。
所以(p_a)对答案贡献了(frac{e_a(e_a+1)}{2} cdot frac{prodlimits_{i=1}^{k}e_i+1}{e_a+1})次,也可以继续化简得到(frac{e_aprodlimits_{i=1}^{k}e_i+1}{2})

指数太大,根据费马小定理,我们可以对(MOD-1)取模。
指数第一部分有个分母(2),由于不能计算(MOD-1)的逆元,所以可以先对(2(MOD-1))取模。
第二部分可以通过计算前缀后缀积来避免除法。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long LL;
const LL MOD = 1000000007;
const int maxn = 100000 + 10;

LL pow_mod(LL a, LL n) {
	LL ans = 1;
	while(n) {
		if(n & 1) ans = ans * a % MOD;
		a = a * a % MOD;
		n >>= 1;
	}
	return ans;
}

int pcnt, prime[maxn], pos[maxn];
bool vis[maxn];

void preprocess() {
	for(int i = 2; i < maxn; i++) {
		if(!vis[i]) prime[++pcnt] = i;
		for(int j = 1; j <= pcnt && i * prime[j] < maxn; j++) {
			vis[i * prime[j]] = true;
			if(i % prime[j] == 0) break;
		}
	}
	for(int i = 1; i <= pcnt; i++) pos[prime[i]] = i;
}

int n;
LL p[maxn];
LL pre[maxn], suf[maxn];

int main()
{
	preprocess();

	while(scanf("%d", &n) == 1) {
		memset(p, 0, sizeof(p));

		for(int i = 1; i <= n; i++) {
			int a; scanf("%d", &a);
			if(i == 1 || a == 0) continue;
			int t = i;
			for(int j = 1; j <= pcnt; j++) {
				if((LL) prime[j] * prime[j] > t) break;
				if(t % prime[j] == 0) {
					int cnt = 0;
					while(t % prime[j] == 0) {
						t /= prime[j];
						cnt++;
					}
					p[j] += (LL)cnt * a;
					p[j] %= 2 * (MOD - 1);
				}
			}
			if(t > 1) { p[pos[t]] += a; p[pos[t]] %= 2 * (MOD - 1); }
		}

		int tot = pcnt;
		while(tot > 1 && p[tot] == 0) tot--;
		pre[0] = suf[tot+1] = 1;
		for(int i = 1; i <= tot; i++) pre[i] = pre[i-1] * (p[i] + 1) % (MOD - 1);
		for(int i = tot; i >= 1; i--) suf[i] = suf[i+1] * (p[i] + 1) % (MOD - 1);

		LL ans = 1LL;
		for(int i = 1; i <= tot; i++) {
			LL ta, tb;
			if(p[i] % 2 == 0) ta = p[i]/2%(MOD-1)*((p[i]+1)%(MOD-1))%(MOD-1);
			else ta = (p[i]+1)/2%(MOD-1)*(p[i]%(MOD-1))%(MOD-1);
			tb = pre[i-1] * suf[i+1] % (MOD - 1);
			ta = ta * tb % (MOD - 1);
			ans = ans * pow_mod(prime[i], ta) % MOD;
		}

		printf("%lld
", ans);
	}

	return 0;
}
原文地址:https://www.cnblogs.com/AOQNRMGYXLMV/p/4933765.html