POJ1845——A的B次方的因子和

算术基本定理

根据算术基本定理，任何一个大于1的自然数n，能够唯一表示为

$$n=\prod_i{p_i}^{r_i}$$
其中， $p_i$是n的第 $i$个质因子。一般将质因子按升序排列，则上述表示形式是唯一的。例如：$12=2^2\times3^1$

因子个数

$n$的所有的因子的个数为：

$$\prod_i(1+r_i)$$
例如12的因子个数为$(2+1)\times(1+1)=6$。

因子和

$n$的所有因子的和为：

$$\prod_i{\sum_{j=0}^{r_i}{p_i^j}}=(1+p_1+p_1^2+...+p_1^{r_1})(1+p_2+p_2^2+...+p_2^{r_2})...$$
注意到右边的每一项其实是等比数列，有

$$\prod_i{\sum_{j=0}^{r_i}{p_i^j}}=\prod_i{{p_i^{r_i+1}-1}\over{p_i-1}}$$

除法在模M的意义下可以转为与其逆元的乘法，但前提是逆元存在。一般而言，题目均要求对质数取模（例如本题）。逆元存在的前提就是$p_i-1$不是M的整数倍，即$p_i\%M\neq1$

如果逆元不存在，则$p_i=kM+1$，所以

$$\sum_{j=0}^{r_i}{p_i^j}=\sum_{j=0}^{r_i}(1+kM)^j$$
在模M的意义下， $右边=(r_i+1)\%M$

POJ1845只需找出A的所有质因子$p_i$及其$r_i$，再将$r_i$乘以B用于计算即可。

#include <iostream>
#include <cstdlib>
#include <time.h>
using namespace std;

typedef long long int llt;
typedef llt int_t;

llt const MOD = 9901LL;

//非递归的扩展欧几里德算法
//返回a、b的gcd，同时x、y满足ax+by=gcd
int_t exEuclid(int_t a,int_t b,int_t&x,int_t&y){
    int_t x0 = 1, y0 = 0;
    int_t x1 = 0, y1 = 1;
    x = 0; y = 1;
    int_t r = a % b;
    int_t q = ( a - r ) / b;
    while( r ){
        x = x0 - q * x1;
        y = y0 - q * y1;
        x0 = x1; y0 = y1;
        x1 = x; y1 = y;
        a = b; b = r; r = a % b;
        q = ( a - r ) / b;
    }
    return b;
}

//求a相对于p的逆元，a、p互质才存在逆元
int_t inv(int_t a,int_t p){
    int_t x,y;
    int_t r = exEuclid(a,p,x,y);
    if ( r != 1 ) return 0;
    x = x % p;
    if ( x < 0 ) x += p;
    return x;
}

//利用二进制计算a*b%MOD
llt multiMod(llt a,llt b,llt MOD){
    llt ret = 0LL;
    a %= MOD;
    while( b ){
        if ( b & 1LL ) ret = ( ret + a ) % MOD, --b;
        b >>= 1LL;
        a = ( a + a ) % MOD;
    }
    return ret;
}

//计算a^b%MOD
llt powerMod(llt a,llt b,llt MOD){
    llt ret = 1LL;
    a %= MOD;
    while( b ){
        if ( b & 1LL ) ret = multiMod(ret,a,MOD),--b;
        b >>= 1LL;
        a = multiMod(a,a,MOD);
    }
    return ret;
}

//质因子p，个数k
llt sum(llt p,llt k){
    if ( 1LL == p % MOD ) return ( k + 1LL ) % MOD;

    llt ret = ( powerMod(p,k+1LL,MOD) - 1LL ) % MOD;
    if ( ret < 0 ) ret += MOD;

    llt i = inv(p-1LL,MOD);
    return multiMod(ret,i,MOD);
}

llt proc(llt a,llt b){
    llt ret = 1LL;

    for(llt i=2LL;i*i<=a;i+=1LL){
        if ( a % i ) continue;

        llt k = 0;
        while( 0LL == a % i ) a /= i, ++k;
        k *= b;
        ret = ( ret * sum(i,k) ) % MOD;
    }

    if ( a != 1LL ) ret = ( ret * sum(a,b) ) % MOD;
    return ret;
}

int main(){
    llt a,b;
    while(cin>>a>>b) cout<<proc(a,b)<<endl;
    return 0;
}

该题的数据实际上没有$A=0$的情况。所以有些AC代码在这时的输出实际上是不一致的。