题解 poj1845 Sumdiv (数论) (分治)

传送门

---

大意：求A^B的所有因子之和，并对其取模 9901再输出

（这题又调了半天，把n和项数弄混了QAQ）

根据算数基本定理：A=(p1^k1)*(p2^k2)*(p3^k3)*...*(pn^kn) （pi为素数）

则A的所有因子之和Sum=(1+p1+p1^2+p1^3+...p1^k1) * (1+p2+p2^2+p2^3+….p2^k2) * (1+p3+ p3^3+…+ p3^k3) * .... * (1+pn+pn^2+pn^3+...pn^kn)

那么A^B = p1^(k1*B) * p2^(k2*B) * p3^(k3*B) *...* pn^(kn*B)

所以Sum=[1+p1+p1^2+...+p1^(a1*B)] * [1+p2+p2^2+...+p2^(a2*B)] *...* [1+pn+pn^2+...+pn^(an*B)]

用分治求等比数列1+pi+pi^2+pi^3+...+pi^n：

（1）若n为奇数（有偶数项）则：
      　　1 + p + p^2 + p^3 +...+ p^n

　　= (1+p^(n/2+1)) + p * (1+p^(n/2+1)) +...+ p^(n/2) * (1+p^(n/2+1))

　　= (1 + p + p^2 +...+ p^(n/2)) * (1 + p^(n/2+1))

（2）若n为偶数（有奇数项）则:
      　　1 + p + p^2 + p^3 +...+ p^n

　　= (1+p^(n/2+1)) + p * (1+p^(n/2+1)) +...+ p^(n/2-1) * (1+p^(n/2+1)) + p^(n/2)

　　= (1 + p + p^2 +...+ p^(n/2-1)) * (1+p^(n/2+1)) + p^(n/2);

可以递归求解

#include<cstdio>
#include<iostream>
#include<algorithm>
#define R register int
#define ll long long
const int mod=9901;
using namespace std;
int a,b,tot,ans=1;
int pm[50010],cnt[50010];

inline ll q_pow(ll a,ll p)
{
    if(a==0) return 1;
    R ret=1; a%=mod;
    for(;p;p>>=1,(a*=a)%=mod) if(p&1) (ret*=a)%=mod;
    return ret;
}

ll sum(ll a,ll c)
{
    if(c==0) return 1;
    if(c&1) return ((1+q_pow(a,(c>>1)+1))%mod*sum(a,(c>>1))%mod)%mod;
    return ((1+q_pow(a,(c>>1)+1))%mod*sum(a,(c-1)>>1)+q_pow(a,c>>1)%mod)%mod;
}

int main()
{
    scanf("%d%d",&a,&b);
    for(R i=2;i*i<=a;i+=(i&1?2:1)) 
        if(a%i==0)
        {
            pm[++tot]=i;
            while(a%i==0) cnt[tot]++,a/=i;
        }
    if(a!=1) pm[++tot]=a,cnt[tot]++;//要判断一下A本身是不是素数
    for(R i=1;i<=tot;i++) ans=(ans*(sum(pm[i],cnt[i]*b)%mod))%mod;
    printf("%d
",ans);
}

如有错误，恳请您指正（我太菜了）；如有不理解，可留言，我会尽量回复。。。（高中生(逃)。。）

by Jackpei 2019.2.25