约数之和(公式法)

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

请你求出S mod 9901的值是多少。

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

输出格式
输出一个整数，代表S mod 9901的值。

数据范围
0≤A,B≤5×107

输入样例：
2 3

输出样例：
15
注意: A和B不会同时为0。

公式：
当b为偶数时，a(a ^ 0 + a ^ 1 + … + a ^ b - 1) + 1
当b为奇数时，把它分半
一半是 (a^0 + a ^1 + … + a ^ b / 2)
另一半是 (a^(b / 2 + 1) + … + a ^ b);
另一半也是 (a^(b / 2 + 1) * (a ^ 0 + … + a ^ (b / 2)));

而在编码中，用sum(a, b) 来体现连加

#include <iostream>
using namespace std;

const int mod = 9901;
int a, b;

int qmi(int a, int b)
{
    a %= mod;
    int res = 1;
    while(b)
    {
        if(b & 1) res = res * a % mod;
        b >>= 1;
        a = a * a % mod;
    }
    return res;
}

int sum(int a, int b)
{
    if(!b) return 1;//纯递归出口
    if(b % 2 == 0) return (a % mod * sum(a, b - 1) % mod + 1) % mod;
    return sum(a, b / 2) % mod * (qmi(a, b / 2 + 1) + 1) % mod;
}

int main(void)
{
    cin >> a >> b;
    int res = 1;
    for(int i = 2; i <= a; i ++ )
    {
        int s = 0;
        while(a % i == 0)
        {
            a /= i;
            s ++ ;
        }
        if(s) res = res * sum(i, s * b) % mod;
    }
    
    if(!a) res = 0;
    cout << res << endl;
    return 0;
}