【ACWing】97. 约数之和

题目地址：

https://www.acwing.com/problem/content/99/

假设现在有两个自然数$a$和$a$，$s$是$a^b$的所有约数之和。请你求出$s\mathrm{m}\mathrm{o}\mathrm{d}9901$的值是多少。

输入格式：
在一行中输入用空格隔开的两个整数$a$和$b$。

输出格式：
输出一个整数，代表 $s\mathrm{m}\mathrm{o}\mathrm{d}9901$的值。

数据范围：
$0\le a,b\le 5\times 10^7,a+b>0$

约数之和有公式，参考https://blog.csdn.net/qq_46105170/article/details/113815655。对于一个正整数$x$，若其质因子分解是$x=p_1^{{\alpha }_1}{p}_2^{{\alpha }_2}...p_k^{{\alpha }_k}$，那么$x$的约数个数是$(1+p_1+p_1^2+...+p_1^{{\alpha }_1})(1+p_2+p_2^2+...+p_2^{{\alpha }_2})...(1+p_k+p_k^2+...+p_k^{{\alpha }_k})$。那么若$a=p_1^{{\alpha }_1}...p_k^{{\alpha }_k}$，那么$a^b$的约数个数是$(1+p_1+...+p_1^{b{\alpha }_1})(1+p_2+...+p_2^{b{\alpha }_2})...(1+p_k+...+p_k^{b{\alpha }_k})$。所以可以先对$a$进行质因子分解，然后问题变为怎么计算$1+p+...+p^{k-1}$。设$f(k)=1+p+...+p^{k-1}$，如果$k$是偶数，则$$\begin{gathered} f(k)=(1+p+...+p^{\frac{k}{2}-1})+(p^{\frac{k}{2}}+p^{\frac{k}{2}+1}+...+p^{k-1}) \\ =(1+p^{\frac{k}{2}})(1+p+...+p^{\frac{k}{2}-1})=(1+p^{\frac{k}{2}})f(\frac{k}{2}) \end{gathered}$$又有$f(1)=1$，所以问题转化为怎么求$p^{\frac{k}{2}}\mathrm{m}\mathrm{o}\mathrm{d}9901$（$9901$是素数），这可以用快速幂来求，参考https://blog.csdn.net/qq_46105170/article/details/113823747。如果$k$是奇数，那么$k-1$是偶数，有$$f(k)=1+p(1+p+...+p^{k-2})=1+pf(k-1)$$就转化为已解决的问题了。代码如下：

#include <iostream>
#include <cstring>
using namespace std;

const int MOD = 9901;

// 快速幂
int qmi(int a, int k) {
    a %= MOD;
    int res = 1;
    while (k) {
        if (k & 1) res = res * a % MOD;
        a = a * a % MOD;
        k >>= 1;
    }
    
    return res;
}

int sum(int p, int k) {
    if (k == 1) return 1;
    if (k % 2 == 0) return (1 + qmi(p, k >> 1)) * sum(p, k >> 1) % MOD;
    return (sum(p, k - 1) + qmi(p, k - 1)) % MOD;
}

int main() {
    int a, b;
    scanf("%d%d", &a, &b);

    int res = 1;
    for (int i = 2; i <= a / i; i++)
        if (a % i == 0) {
            int cnt = 0;
            while (a % i == 0) {
                cnt++;
                a /= i;
            }
            
            res = res * sum(i, b * cnt + 1) % MOD;
        }

    if (a > 1) res = res * sum(a, b + 1) % MOD;
    if (a == 0) res = 0;

    printf("%d\n", res);

    return 0;
}

时间复杂度$O(\sqrt{a}+{\log }^{2}b)$，空间$O(\log b)$。