【POJ 1845】sumdiv约数和

FZOJ1831: 【POJ 1845】sumdiv约数和

FZOJ传送门

首先，我们得知道A的约数等于什么
$$A=p_1^{k_1}\ast p_2^{k_2}\ast \cdots \ast p_n^{k_n}$$其中p_n为质数，k_n为任意整数。
根据幂的运算，易得
$$A^B=p_1^{k_{1}B}\ast p_2^{k_{2}B}\ast \cdots \ast p_n^{k_{n}B}$$其次，我们还有一个约数和公式
$$S=(1+p_1+p_1^2+\cdots +p_1^{k_1})\ast (1+p_2+p_2^2+\cdots +p_2^{k_1})\cdots$$根据等比数列以及上面的公式知：
$$S=(\frac{1-p_1^{k_{1}B+1}}{1-p_1})\ast (\frac{1-p_2^{k_{2}B+1}}{1-p_2})\cdots$$
即:
$$S=(\frac{p_1^{k_{1}B+1}-1}{p_1-1})\ast (\frac{p_2^{k_{2}B+1}-1}{p_2-1})\cdots$$

直接暴力求解等比数列可能会超时，所以很自然地想到了逆元（费马小定理），但是此时有一点要注意：p_i-1可能是9901的倍数，就会不存在逆元。
对于$\frac{A}{B}\equiv x(modm)$
等价于：
$$\frac{A}{B}+km=x$$通分得：
$$\frac{A+kBm}{B}=x$$即：
$$\frac{A(modBm)}{B}=x$$
应用这个公式，用到快速幂取模，在快速幂取模中一步乘法可能会溢出long long ,因此有添加了一个对于大数a*b (mod m)的函数。因为最终的模数不是mod，而是$mod\ast (k_i-1)$；

贴代码

#include<bits/stdc++.h>
typedef long long ll;
using namespace std;

const int mac=50009;
const int MOD=9901;
ll B,n,num;

struct node{
    ll p,k;
}yin[mac];

inline ll mul(ll a,ll b,ll mod){//防止爆long long
    ll s=0;
    while(b){
        if(b&1)
            s=(s+a)%mod;
        a=(a*2)%mod;
        b>>=1;
    }
    return s<0?s+mod:s;
}

inline ll mi(ll a,ll b,ll mod){
    ll sum=1;
    while(b>0){
        if(b&1) sum=mul(sum,a,mod);
        a=mul(a,a,mod);
        b>>=1;
    }
    return sum<0?(sum+mod):sum;
}

int main(){
    // freopen("1831.in","r",stdin);
    scanf("%lld%lld",&n,&B);
    ll A=n;
    num=1;
    for(ll i=2;i<=sqrt(n+0.0);i++){
        if(n%i==0){
            yin[num++].p=i;
            while(n%i==0){
                yin[num-1].k++;
                n/=i;
            }
        }
    }
    if(n>1){//如果a 为素数
            yin[num++].p=n;
            yin[num-1].k++;
        }

    ll ans=1;
    for(ll i=1;i<=num;i++){
        ll b=((yin[i].k)*(B));
        ll p=mi(yin[i].p , b+1, MOD*(yin[i].p-1));
        ans*=(p-1)/(yin[i].p-1);
        ans%=MOD;
    }
    printf("%lld\n",ans);
    return 0;
}

感谢博主Sharp_Edge_Of_Defens的博客，在他的启发下，本人才做出这道题。
参考自 https://blog.csdn.net/u012860428/article/details/41315081