二分求等比数列和模板

num73

已于 2024-04-10 13:02:14 修改

阅读量6

点赞数

分类专栏：算法模板文章标签：算法数据结构

于 2020-10-16 21:26:16 首次发布

本文链接：https://blog.csdn.net/qq_44743100/article/details/109125179

版权

算法模板专栏收录该内容

14 篇文章 0 订阅

订阅专栏

#include <iostream>
#include <string.h>
#include <stdio.h>
 
using namespace std;
const int M = 1000000007;
typedef long long LL;
 
LL power(LL a,LL b)
{
    LL ans = 1;
    a %= M;
    while(b)
    {
        if(b & 1)
        {
            ans = ans * a % M;
            b--;
        }
        b >>= 1;
        a = a * a % M;
    }
    return ans;
}
 
LL sum(LL a,LL n)
{
    if(n == 1) return a;
    LL t = sum(a,n/2);
    if(n & 1)
    {
        LL cur = power(a,n/2+1);
        t = (t + t * cur % M) % M;
        t = (t + cur) % M;
    }
    else
    {
        LL cur = power(a,n/2);
        t = (t + t * cur % M) % M;
    }
    return t;
}
 
int main()
{
    LL a,n;
    while(cin>>a>>n)
        cout<<sum(a,n)<<endl;
    return 0;
}