【CodeForces】678D - Iterated Linear Function

http://codeforces.com/problemset/problem/678/D

G(n)=A*G(n-1)+b
G(0)=x

给定A,b,n,x,求G(n)。

快速幂应用。

#include <bits/stdc++.h>
using namespace std;

long long n,x,a,b;
long long mod=1e9+7;

int main(){

    cin >> a >> b >> n >> x;
    while (n){
        if (n&1) x=(a*x+b)%mod;
        b=b*(a+1)%mod;
        a=a*a%mod;
        n/=2;
    }
    cout << x << endl;
}