NYOJ 640 Geometric Sum

最新推荐文章于 2016-12-10 20:23:18 发布

倚世独殇

最新推荐文章于 2016-12-10 20:23:18 发布

阅读量797

点赞数

分类专栏：快速幂(模) 文章标签： NYOJ 640 Geometric S

本文链接：https://blog.csdn.net/u012804490/article/details/25039221

版权

快速幂(模) 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

Geometric Sum

时间限制： 1000 ms | 内存限制： 65535 KB

难度： 3

描述

Compute (a + a^2 + … + a^n) mod m.

输入

Three integers a,n,m.
(1≤a,n,m≤10^18)
It ends with EOF.

输出

The only integer denotes the result.

样例输入

2 2 1000000000

样例输出

别人的AC码： 
#include <stdio.h>
typedef long long LL;
LL Mod;
LL multMod(LL a,LL b)
{
    LL res = 0,base = a;
    while(b)
    {
        if(b&1)
            (res += base) %= Mod;
        (base <<= 1) %= Mod;
        b >>= 1;
    }
    return res;
}

LL powerMod(LL a,LL n)
{
    LL res = 1 , base = a;
    while(n)
    {
        if(n&1)
            res = multMod(res,base);
        base = multMod(base,base);
        n >>= 1;
    }
    return res;
}

LL solve(LL a,LL n)
{
    if(n == 1)
        return a;
    LL halfsum = solve(a,n >>1);
    if(n&1)
    {
        LL half = powerMod(a,n+1 >>1);
        return (halfsum + half + multMod(half,halfsum)) % Mod;
    }
    else
    {
        LL half = powerMod(a,n >>1);
        return (halfsum + multMod(half,halfsum)) % Mod;
    }
}

int main()
{
    LL a,n;
    while(~scanf("%lld%lld%lld",&a,&n,&Mod)){
    a %= Mod;
    printf("%lld\n",solve(a,n));
    }
	return 0;
}