【蓝皮书】T97约数之和

题目链接

一、题目分析

题目要求$A^B$的所有因数之和，数据范围很大，用暴力不太合适。
本题应用到一些数学知识

• 算数基本定理：任何大于1的正整数都能唯一分解为有限个质数的乘积，可写作:$$N=p_1^{c_1}{p}_2^{c_2}{p}_3^{c_3}\dots p_m^{c_m}$$其中$c_i$都是正整数，$p_i$都是质数，且满足$p_1<p_2<\dots <p_m$
  那么可以将输入的A进行质因数分解，
  $$A=p_1^{c_1}{p}_2^{c_2}{p}_3^{c_3}\dots p_m^{c_m}$$
  同样有
  $$A^B=p_1^{c_1\ast B}{p}_2^{c_2\ast B}{p}_3^{c_3\ast B}\dots p_m^{c_m\ast B}$$
  $A^B$的因数集合可以表示为 { p 1 c 1 ∗ p 2 c 2 ∗ p 3 c 3 ∗ … ∗ p m c m } \{p_1^{c_1}*p_2^{c_2}*p_3^{c_3}*{\ldots}*p_m^{c_m}\} {p1c1​​∗p2c2​​∗p3c3​​∗…∗pmcm​​}其中$0⩽k_i⩽B\ast c_i(1⩽i⩽n)$
  利用乘法分配律（类似组合的过程，从每个括号里选一个数，然后乘起来），可以将因数相加转化为多组乘法

• $$(1+p_1^1+p_1^2+\dots +p_1^{c_1\ast B})\ast (1+p_2^1+p_2^2+\dots +p_1^{c_1\ast B})\ast \dots \ast (1+p_n^1+p_n^2+\dots +p_n^{c_n\ast B})$$

每个括号中的内容均为等比数列，等比数列求和可以采用公式法，但是，对除法运用模运算不存在分配性（可以通过逆元解决，这里先不进行讨论）。考虑采用分治法的思想解决问题。所谓分治法即将问题分解为多个子问题，对子问题求解后，再合并为原问题。等比数列求和：
$$sum(p,c)=1+p^1+p^2+\dots +p^c$$
若c为奇数：
$$sum(p,c)=(1+p^1+p^2+\dots +p^{(c-1)/2})+(p^{(c+1)/2}+\dots +p^c)$$
$$=(1+p+\cdots +p^{(c-1)/2})+p^{(c+1)/2}\ast (1+p+\cdots +p^{(c-1)/2})$$
$$=(1+p^{(c+1)/2})\ast sum(p,(c-1)/2)$$
若c为偶数，同理可得：
$$sum(p,c)=(1+p^{c/2})\ast sum(p,c/2-1)+p^c$$

二、代码

#include<bits/stdc++.h>
using namespace std;
long long m = 9901;
int pow(int a,int b)
{
    int ans = 1 % m;
    for(; b; b >>= 1){
        if(b & 1) ans = (long long)ans * a % m;
        a = (long long)a * a % m;
    }
    return ans;
}
int sum(int p,int c)//分治法计算等比数列和
{
    if(c == 0) return 1;
    if(c % 2 == 0){
        return ((1 + pow(p,c >> 1)) * sum(p,(c >> 1) - 1) + pow(p,c)) % m;
    }
    else if(c % 2 == 1){
        return (sum(p,(c-1) >> 1) * (pow(p,(c + 1) >> 1) + 1)) % m;
    }
}
int main()
{
    int a,b;
    cin >> a >> b;
//    cout << pow(a,b) << endl;
//    cout << sum(a,b) << endl;
    if(a == 0){
        cout << 0 << endl;
        return;
    }
    int ans = 1;
    for(int i = 2;i <= a;++i){
        int j = 0;
        while(a % i == 0){
            ++j;
            a/=i;
        }
//        cout << i << ' ' << j << endl;
        ans *= sum(i,j * b);
        ans = ans % m;
    }
    cout << ans << endl;
}