poj 1845 Sumdiv

3，我们在求解：

是应用了二分等比数列求和法

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
using namespace std;
const int mod=9901;
const int Max=7100;
typedef long long LL;
int A,B;
int sum[Max];//A 分解的质因数对应的次数
int data[Max];//A 分解的质因数
int length;

int prime[Max],Size;
bool used[Max];

void init()
{//素数打表
memset(used,true,sizeof(used));
Size=0;
for(int i=2;i<Max;i++)
{
if(used[i])
prime[Size++]=i;
for(int j=0;j<Size && prime[j]*i<Max;j++)
{
used[i*prime[j]]=false;
if(i%prime[j]==0)
break;
}
}
}

void draw()
{//A 分解质因数
int key=A;
length=0;
int bian=sqrt(A);
for(int i=0; prime[i]<=bian && key>1; i++)
{
int x=0;
while(key%prime[i]==0)
{
x++;
key=key/prime[i];
}
if(x)
{
data[length]=prime[i];
sum[length]=x*B;
length++;
}
}
if(key!=1)
{
data[length]=key;
sum[length]=B;
length++;
}
}
LL pow(int n,int x)
{//快速幂运算
LL res=1;
LL y=x;
while(n)
{
if(n&1)
res=(res*y)%mod;
y=(y*y)%mod;
n>>=1;
}
return res%mod;
}

LL DFS(int n,int x)
{//二分求等比数列和
if(n==0)
return 1;
LL res,key;
if(n%2==0)
{
key=pow(n/2,x);
res=(key*x%mod)%mod+1;
res=((res*DFS(n/2-1,x))%mod+key)%mod;
}
else
{
res=pow(n/2+1,x)+1;
res=(res*DFS(n/2,x))%mod;
}
return res%mod;
}
void solve()
{
LL res=1;
for(int i=0;i<length;i++)
{
res=(res*DFS(sum[i],data[i]))%mod;
}
printf("%ld\n",res);
}
int main()
{
init();
while(scanf("%d%d",&A,&B)!=EOF)
{
draw();
solve();
}
return 0;
}