首页 > 代码库 > NYOJ 640 Geometric Sum
NYOJ 640 Geometric Sum
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
- 样例输出
6
别人的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; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。