POJ - 1845 G - Sumdiv (唯一分解定理)

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#define mod 9901
#define MAX 1001
using namespace std;
typedef long long ll;
int cnt;
vector<ll> prime;
vector<ll> times;
ll qpow(ll a,ll b)//快速幂加快求幂数
{
    ll ans=1;
    while(b)
    {
        if(b&1) ans=(ans*a)%mod;
        b>>=1;
        a=(a*a)%mod;
    }
    return ans;
}
void divide(ll n) {//求出n里所有的素数银子
    for(ll i=2;i*i<=n;++i) {
        if(n%i==0) {
            prime.push_back(i);ll cnt=0;
            while(n%i==0) {n/=i;++cnt;}
            times.push_back(cnt);
        }
    }
    if(n>1) {prime.push_back(n);times.push_back(1);}
}
int main()
{
    ll a,b;
    scanf("%lld%lld",&a,&b);
    divide(a);
    ll ans=1;
    for(int i=0,end=prime.size();i<end;++i) {
        times[i]*=b;//乘了括号外的乘方
        if((prime[i]-1)%mod==0) ans=ans*(times[i]+1)%mod;//当底数为1时，就是乘以项数
        else ans=ans*((qpow(prime[i],times[i]+1)-1+mod)*qpow(prime[i]-1,mod-2)%mod)%mod;//等比数列公式
    }
    printf("%lld\n",ans);
    return 0;
}