Light oj 1054 - Efficient Pseudo Code

题目：Light oj 1054 - Efficient Pseudo Code

思路：打出65536以内的素数，质因素分解后，判断最后剩下的一个数字如果不等于1的话，说明存在大于16位的素因子，这个特殊判断。

剩下的求约数的和，满足的就是求 （1+p+p^2+p^3...+p^n）的累积（这个式子的展开就是约数之和）这个式子用二分求

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
#define mod 1000000007
#define maxn 65540
int vis[maxn];
int prime[maxn];
long long num[maxn/10];
int n_prime=0;
void Prime()
{
    memset(vis,1,sizeof(vis));
    vis[1]=vis[0]=0;
    for(int i=2;i<maxn;i++)
        if(vis[i])
        {
            prime[++n_prime]=i;
            for(int j=2*i;j<maxn;j+=i)
                vis[j]=0;
        }
    //printf("%d  %d\n",n_prime,prime[n_prime]);
}
long long Pow(long long a,long long b)
{
    long long ans=1;
    while(b)
    {
        if(b&1)
        {
            b--;
            ans=(ans*a)%mod;
        }
        else
        {
            b/=2;
            a=(a*a)%mod;
        }
    }
    return ans;
}
long long sum(long long p,long long n)
{
     if(n==0)
        return 1;
     if(n==1)
        return p+1;
     if(n&1)
     {
         return (sum(p,n/2)*((1+Pow(p,n/2+1))%mod))%mod;
     }
     else
     {
         return ((sum(p,n/2-1)*((1+Pow(p,n/2))%mod))%mod+Pow(p,n))%mod;
     }
}
int main()
{
    Prime();
    int t;
    scanf("%d",&t);
    for(int cases=1;cases<=t;cases++)
    {
        int m,n;
        scanf("%d%d",&n,&m);
        memset(num,0,sizeof(num));
        for(int i=1;i<=n_prime&&prime[i]<=n;i++)
        {
            if(n%prime[i]==0)
            {
                while(n%prime[i]==0)
                {
                    n/=prime[i];
                    num[i]+=m;
                }
            }
        }
        long long ans=1;
        if(n!=1)
        {
            ans=sum(n,m);
        }
        long long tmp;
        for(int i=1;i<=n_prime;i++)
        {
            tmp=sum(prime[i],num[i]);
            ans=(ans*tmp)%mod;
        }
        printf("Case %d: ",cases);
        printf("%lld\n",ans);
    }
    return 0;

}