题解：P1593 因子和

因子和

题目描述

输入两个整数 $a$ 和 $b$，求 $a^b$ 的因子和。

由于结果太大，只要输出它对 $9901$ 取模的结果。

输入格式

仅一行，为两个整数 $a$ 和 $b$。

输出格式

输出一行一个整数表示答案对 $9901$ 取模的结果。

样例 #1

样例输入 #1

2 3

样例输出 #1

15

提示

数据规模与约定

对于全部的测试点，保证 $1\le a\le 5\times 10^7$，$0\le b\le 5\times 10^7$。

---

个人题解

思路

给定 $a,b$ 求 $a^b$ 这个数所有的因数和，答案对 9901 取模。

思路

由唯一分解定理得任意正整数

$Z^{+}=p_1^{k^1}\ast p_2^{k^2}\ast p_3^{k^3}\ast ....\ast p_n^{k^n}$

又由唯一分解定理得一个正整数的因数和

$s=(1+p_1^1+p_1^2+p_1^3+....+p_1^{k^1})\ast (1+p_2^1+p_2^2+p_2^3+....+p_2^{k^2})\ast ...\ast (1+p_n^1+p_n^2+p_n^3+....+p_n^{k^n})$

我们知道两个同底数次幂相乘，指数相加，即

$a^b\ast a^c=a^{b+c}$

所以当我们要求 $a^b$ 的所有因数的和时，不妨先考虑底数 $a$ 的分解，当我们得到 $a$ 的分解后，对于任意的 $p_i^{k_i}$ 只需要视为 $p_i^{b\ast k_i}$ 即可，但由于 $b\le 5\ast 10^7$ 所以我们可以考虑欧拉降幂，即

$a^b=a^{b\mathrm{mod}\varphi (p)+\varphi (p)}(\mathrm{mod}p)$

分析完毕。

代码

#include <iostream>
#include <vector>
using namespace std;
#define ll long long
const ll MOD = 9901;
// p 为质数，所以phi(p) = p - 1
const ll phi = 9900;
struct Prim {
	ll p, cnt;
};
ll quick_mi(ll a, ll b) {
	ll res = 1;
	while (b) {
		if (b % 2) res = res * a % MOD;
		a = a * a % MOD;
		b /= 2;
	}
	return res;
}
vector<Prim> ans;
void fenjie(ll n) {
	for (ll x = 2; x * x <= n; x++) {
		if (n % x) continue;
		ll cnt = 0;
		while (n % x == 0) cnt++, n /= x;
		ans.push_back({ x, cnt });
	}
	if (n > 1) ans.push_back({ n, 1 });
}
int main() {
	ll a, b, s = 1;
	cin >> a >> b;
	if (a == 1 || b == 0) {
		cout << 1;
		return 0;
	}
	fenjie(a);
	for (auto& it : ans) {
		ll temp = 1;
		it.cnt = it.cnt * b;
		it.cnt = it.cnt >= phi ? it.cnt % phi + phi : it.cnt;
		ll temp_s = 0;
		for (int x = 0; x <= it.cnt; x++) {
			temp_s = (temp_s + temp) % MOD;
			temp = temp * it.p % MOD;
		}
		s = s * temp_s % MOD;
	}
	printf("%lld\n", s % MOD);
	return 0;
}