Light OJ 1054 - Efficient Pseudo Code

思路：

主要是分解质因子，然后可以推出，所有因子的和 等于 （2^0 + 2^1 +2^2 + …… 2^p1）*(3^0 + 3^1 + 3^2 +…… 3^p2) * (5^0 + 5^1 + 5^2 + 5^3 + …… +5^p3)*…………

即所有质因子前p项和的乘积。最后还要注意等比数列 1 + x + x^2 + ...+ x^t的求法，利用递推。

#include <stdio.h>
#include <iostream>
#include <string>
#include <string.h>
#include <algorithm>
#include <stdlib.h>
#include <math.h>
#include <vector>
#include <map>

using namespace std;
typedef long long ll;
const int maxn = 1000005;
const int mod = 1000000007;
bool isprime[maxn];
int prime[maxn];
void getprime(){
    memset(isprime, 0, sizeof(isprime));
    int id = 1;
    for(int i = 2;i < maxn;i++){
        if(!isprime[i])
            prime[id++] = i;
        for(int j = i+i;j < maxn;j += i)
            if(!isprime[j])
                isprime[j] = 1;
    }
    //    for(int i = 1;i < 100;i++)
    //        printf("%d ",prime[i]);
}
vector<pair<ll, ll> >factor;

void primeFactor(int n,int m){
    for(int i = 1;(ll)prime[i]*prime[i] <= (ll)n;i++){
        int p = 0;
        while(n%prime[i] == 0){
            p++;
            n /= prime[i];
        }
        if(p)
            factor.push_back(make_pair((ll)prime[i], (ll)p*m));
    }
    if(n != 1)
        factor.push_back(pair<ll, ll>((ll)n,(ll)m));
}
ll pow(ll x,ll y){
    ll ans = 1;
    while(y){
        if(y&1){
            ans *= x;
            ans %= mod;
        }
        x *= x;
        x %= mod;
        y >>= 1;
    }
    return ans;
}
//最好不要推出前n项和的公式然后再计算，因为有除法，也可以求逆元利用前n项和的公式。
/*
 求sum=x^1+x^2+x^3+x^4+x^5+x^6+x^7 ....... x^t
 共有t项
 公式:
 若t%2==0     sum(t)=sum(t/2)+sum(t/2)*2^(t/2);
 若t%2==1     sum(t)=sum(t/2)+sum(t/2)*2^(t/2)+ x^t;
 */

ll  sum(ll x , ll t){ // 等比数列 1 + x + x^2 + ...+ x^t
    if(t == 0)  return 1 ;
    ll ans;
    if(t&1)
        ans = sum(x , t>>1) * (1 + pow(x , (t>>1) +1)) % mod ;
    else
        ans = (sum(x , (t>>1)-1) * (1 + pow(x , t>>1)) % mod) + pow(x , t) ;
    return ans;
}
ll  answer(){
    ll ans = 1;
    pair<ll, ll>pp;
    for(int i = 0;i < factor.size();i++){
        pp = factor[i];
        ans *= sum(pp.first, pp.second);
        ans %= mod;
    }
    return ans ;
}

int main(){
    getprime();
    int t,n,m;
    scanf("%d",&t);
    for(int cas = 1;cas <= t;cas++){
        scanf("%d%d",&n,&m);
        factor.clear();
        primeFactor(n,m);
        printf("Case %d: %lld\n",cas,answer());
        
//        pair<int, int>pp;
//        for(int i = 0;i < factor.size();i++){
//            pp = factor[i];
//            printf("%d %d\n",pp.first,pp.second);
//        }
    }
    return 0;
}