洋灰三角

链接：https://www.nowcoder.com/acm/contest/136/J
来源：牛客网
 

题目描述

    洋灰是一种建筑材料，常用来筑桥搭建高层建筑，又称，水泥、混凝土。

    WHZ有很多铸造成三角形的洋灰块，他想把这些洋灰三角按照一定的规律放到摆成一排的n个格子里，其中第i个格子放入的洋灰三角数量是前一个格子的k倍再多p个，特殊地，第一个格子里放1个。
    WHZ想知道把这n个格子铺满需要多少洋灰三角。

输入描述:

第一行有3个正整数n，k，p。

输出描述:

输出一行，一个正整数，表示按照要求铺满n个格子需要多少洋灰三角，由于输出数据过大，你只需要输出答案模1000000007(1e9+7)后的结果即可。

示例1

输入

复制

3 1 1

输出

复制

6

说明

洋灰三角铺法：1 2 3，总计6个

示例2

输入

复制

3 2 2

输出

复制

15

说明

洋灰三角铺法：1 4 10，总计15个

示例3

输入

复制

3 3 3

输出

复制

28

说明

洋灰三角铺法：1 6 21，总计28个

再次遇到了  a[n]=k*a[n-1]+p， 让我们来求通项公式的题。 

这种题，我们知道它是等比数列，所以我们直接  a[n]+t = k(a[n-1]+t) ，然后就说明 k*t -t=p ,

所以这个t就解出来了。 剩下的就简单了，求个逆元。

#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=a;i<b;++i)
#define per(i,a,b) for(int i=b-1;i>=a;i--)
#define ls(x) (x<<1)
#define rs(x) (x<<1|1)
typedef long long ll;

const int maxn=2e5+10;
const int mod=1e9+7;

ll pow_mod(ll base,int n,int mod){
    ll ans=1;
    while(n){
        if(n&1)ans=ans*base%mod;
        base=base*base%mod;
        n>>=1;
    }
    return ans%mod;
}

int main(){
    ll inv2=pow_mod(2,mod-2,mod);
    ll n,k,p;
    scanf("%lld %lld %lld",&n,&k,&p);
    if(k==1){
        ll ans1=n*(n-1)%mod*inv2%mod*p%mod;
        ll ans=(ans1+n)%mod;
        printf("%lld\n",ans);
    }
    else{
        ll inv_k=pow_mod(k-1,mod-2,mod);
        ll b1=(1ll+p*inv_k%mod)%mod;
        ll ans1=pow_mod(k,n,mod);
        ans1=(ans1-1+mod)%mod;
        ans1=ans1*inv_k%mod;
        ans1=ans1*b1%mod;
        ll ans2=n*p%mod*inv_k%mod;
        ans1=(ans1-ans2+mod)%mod;
        printf("%lld\n",ans1);
    }
	return 0;
}