【Luogu】P1593因子和（唯一分解定理，约数和公式）

题目链接

首先介绍两个定理。

整数唯一分解定理：任意正整数都有且只有一种方式写出素数因子的乘积表达式。

\(A=(p1^k1 p2^k2 ...... pn^kn \)

求这些因子的代码如下

for(int i=;i*i<=a;++i){
        if(!(a%i)){
            prime[++num]=i;
            while(!(a%i)){
                a/=i;
                sum[num]++;
            }
        }
    }
    if(a!=){
        prime[++num]=a;
        sum[num]=;
    }

唯一分解定理

约数和公式：对于已经分解的整数A，有A的所有因子和为

\( S= (1+p1+p1²+p1³+......+p1^k1) (1+p2+p2²+p2³+......+p2^k2)........(1+pn+pn²+pn³+......+pn^kn) \)

所以局势明朗。用快速幂求出p的k*b次方，然后递归求和。代码如下

long long Sum(long long p,long long n){
    if(n==)    return ;
    if(n&)    return (Sum(p,n>>)*(+Pow(p,(n>>)+)))%mod;
    return     (Sum(p,(n>>)-)*(+Pow(p,(n>>)+))+Pow(p,n>>))%mod;
}

解题代码如下

#include<cstdio>
#include<cctype>
#define mod 9901
inline long long read(){
    long long num=,f=;
    char ch=getchar();
    while(!isdigit(ch)){
        if(ch=='-')    f=-;
        ch=getchar();
    }
    while(isdigit(ch)){
        num=num*+ch-'';
        ch=getchar();
    }
    return num*f;
}

int prime[];
int sum[];
int num;

long long Pow(long long n,long long i){
    if(i==)    return ;
    if(i==)    return n%mod;
    long long ret=Pow(n,i>>);
    if(i&)    return (((ret*ret)%mod)*n)%mod;
    return (ret*ret)%mod;
}

long long Sum(long long p,long long n){
    if(n==)    return ;
    if(n&)    return (Sum(p,n>>)*(+Pow(p,(n>>)+)))%mod;
    return     (Sum(p,(n>>)-)*(+Pow(p,(n>>)+))+Pow(p,n>>))%mod;
}

int main(){
    long long a=read(),b=read();
    for(int i=;i*i<=a;++i){
        if(!(a%i)){
            prime[++num]=i;
            while(!(a%i)){
                a/=i;
                sum[num]++;
            }
        }
    }
    if(a!=){
        prime[++num]=a;
        sum[num]=;
    }
    int ans=;
    for(int i=;i<=num;++i)
        ans=(ans*Sum(prime[i],sum[i]*b)%mod)%mod;
    printf("%d",ans);
    return ;
}