Acwing:约数之和

题目：假设现在有两个自然数 A 和 B，S 是 AB 的所有约数之和。

请你求出 Smod9901 的值是多少。

输入格式
在一行中输入用空格隔开的两个整数 A 和 B。

输出格式
输出一个整数，代表 Smod9901 的值。

数据范围
0≤A,B≤5×107
输入样例：
2 3
输出样例：
15
注意: A 和 B 不会同时为 0。

公式和一些原理：
质因数分解，就是将一个数分解成为 p1^c1×p2^c2×…×pn^cn 比如说24=2^3*3^1=8*3（注意是相乘不是相加）
那么
A=p1^k1*p2^k2*......*pn*kn;
A^B=p1^(k1*B)*p2^(k2*B).....;

约数个数：s=(k1+1)(k2+1)(k3+1)...{
这里就有两种情况，
打个比方，k是奇数，k+1则是偶数，
则k是奇数，约数个数是偶数个数
k是偶数，约数个数是奇数个数}

约数之和sum=p^0+p^1+p^2+....

举例：
180=2^2*3^2*5^1
那么上式代入公式，k1=2，k2=2，k3=1，p1=2，p2=3，p3=5

约数个数就等于（2+1）（2+1）（1+1）=18  分别是奇*奇*偶

约数之和=(p1^0+p1^1+..p1^k1)(p2^0+p2^1+..p2^k2)(.....)
             =(2^0+2^1+2^2)(3^0+3^1+3^2)(5^0+5^1)
             =(2+3+5)^0+(2+3+5)^1+(2+3)^2
             =(p^0+p^1+p^2)
这里就可以理解成简便运算，把乘法转成加法。

代码：
#include <iostream>

using namespace std;

const int mod = 9901;

int ksm(int a, int k)//快速幂函数
{
    a %= mod;
    int res = 1;
    while (k!=0)
    {
        if (k & 1) res = res * a % mod;
        a = a * a % mod;
        k >>= 1;
    }
    return res;
}

int sum(int p, int k)
{
    if (k == 0) return 1;
    if (k % 2 == 0)//偶数情况
    return (p % mod * sum(p, k - 1) % mod + 1) % mod;//偶数情况下，约数个数是奇数，+1代表的是先把p^0提出来，让约数个数变为偶数。
    else//奇数情况
    return sum(p, k / 2) % mod * (1 + ksm(p, k / 2 + 1)) % mod;//若k是奇数，代表约数个数是偶数，先把最后一项乘以res，再开始计算前面的所有数
}

int main()
{
    int A, B;
    cin >> A >> B;

    int res = 1;
    for (int i = 2; i <= A; i ++ )
    {
        int s = 0;
        while (A % i == 0)//质因子分解
        {
            s ++ ;
            A /= i;
        }

        if (s!=0) res = res * sum(i, s * B) % mod;
    }

    if (A==0)//注意A=0的情况
    cout<<0<<endl;
    else
    cout << res << endl;
    return 0;
}