【算法刷题】AcWing 97. 约数之和——递推

最新推荐文章于 2024-10-01 23:05:47 发布

Ricky_0528

最新推荐文章于 2024-10-01 23:05:47 发布

阅读量197

点赞数

分类专栏： Algorithm 文章标签：约数之和快速幂因式分解递归算法编程竞赛

本文链接：https://blog.csdn.net/weixin_52665939/article/details/126023059

版权

Algorithm 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

假设现在有两个自然数 $A$ 和 $B$ ， $S$ 是 $A^B$ 的所有约数之和。
请你求出 $S\ mod\ 9901$ 的值是多少。
输入格式
在一行中输入用空格隔开的两个整数 A 和 B。
输出格式
输出一个整数，代表 Smod9901 的值。
数据范围
$\leq A,B \leq 5 \times 10^7$
输入样例

2 3

输出样例

注意： $A$ 和 $B$ 不会同时为 $0$

①分析

$约数相关公式\\ 已知：N=p_1^{\alpha_1}·p_2^{\alpha_2}...p_k^{\alpha_k}\\ 约数个数=(\alpha_1+1)·(\alpha_2+1)...(\alpha_k+1)\\ 约数之和=(p_1^0+p_1+p_1^2+...p_1^{\alpha_1})·(p_2^0+p_2+p_2^2+...p_2^{\alpha_2})...(p_k^0+p_k+p_k^2+...p_k^{\alpha_k})\\$

$本题求的是A^B\\ 可以先将A进行因式分解=p_1^{\alpha_1}·p_2^{\alpha_2}...p_k^{\alpha_k}\\ 则A^B=p_1^{\alpha_1B}·p_2^{\alpha_2B}...p_k^{\alpha_kB}，就可以使用约数之和公式求解\\ 下面要解决的就是p_1^0+p_1+p_1^2+...p_1^{k-1}怎么求，直接算会超时\\ 法一：等比数列求和公式=\frac{p^k-1}{p-1}，分子用快速幂，\frac{1}{p-1}就是求p-1关于9901的逆元\\ 法二：递归=sum(p,k)，看其能否转化为更小的子问题\\ ①k为偶数sum(p,k)=p^0+p^1+...+p^{\frac{k}{2}-1}+p^{\frac{k}{2}}(p^0+p^1+...+p^{\frac{k}{2}-1})=(1+p^{\frac{k}{2}})sum(p,\frac{k}{2})\\ ②k为奇数sum(p,k)=p^0+p(p^0+p^1+...+p^{k-2})=1+p\ sum(p,k-1)=sum(p,k-1)+p^{k-1}$

②代码

#include <cstdio>

const int MOD = 9901;

int qmi(int a, int k)
{
    int res = 1;
    a %= MOD;
    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;
    else if (k % 2 == 0)
        return (1 + qmi(p, k / 2)) * sum(p, k / 2) % MOD;
    else
        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 * i <= a; i++)
        if (a % i == 0)
        {
            int p = 0;
            while (a % i == 0)
            {
                a /= i;
                p++;
            }
            res = res * sum(i, p * b + 1) % MOD;
        }
    
    if (a != 1)
        res = res * sum(a, b + 1) % MOD;
    if (a == 0)
        res = 0;
    
    printf("%d\n", res);
    
    return 0;
}

③细节分析

快速幂

int qmi(int a, int k)
{
    int res = 1;
    a %= MOD; // a可能会很大，a*a就会爆int
    while (k)
    {
        if (k & 1)
            res = res * a % MOD;
        a = a * a % MOD;
        k >>= 1;
    }
    return res;
}

将a分解质因子

int res = 1; // 答案
for (int i = 2; i * i <= a; i++)
    if (a % i == 0)
    {
        int p = 0; // 当前质因子i的最高次数
        while (a % i == 0)
        {
            a /= i;
            p++;
        }
        res = res * sum(i, p * b + 1) % MOD;
    }

a != 1说明还存在一个的质因子a，其次数为1

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

sum(i, p * b + 1)为公式里面 $p_k^{\alpha_kB}=i^{p\ b}$ 这一项，推出的公式中sum的第二个参数为最高次幂加一
递推着求 $p_k^{\alpha_kB}=i^{p\ b}$
```
int sum(int p, int k)
{
    if (k == 1) // k=1即次数为0，值为1
        return 1;
    else if (k % 2 == 0)
        return (1 + qmi(p, k / 2)) * sum(p, k / 2) % MOD;
    else
        return (sum(p, k - 1) + qmi(p, k - 1)) % MOD;
}
```