传送门
题目描述:
Input
共1行,4个整数数p, q, r, n中间用空格分隔(1 <= p, q, r, n<=1000000000)。
Output
对于每一个数据,在一行中输出答案。
Input示例
2 2 1 1
Output示例
3
解题思路:
令
f(n)=∑ni=0ai∗bn−i
,将其展开有:
f(n)=a0∗bn+a1∗bn−1+...+an∗b0
同理
f(n+1)
也可写出:
f(n+1)=∑n+1i=0ai∗bn+1−i
f(n+1)=a0∗bn+1+a1∗bn+...+an+1∗b0
通过观察,发现如下公式成立:
f(n+1)=f(n)∗q+an+1∗b0
根据这个递推式,我们可以构造一个矩阵
A
:
(f(n),an,r)∗A=(f(n+1),an+1,r) ,即有:
(f(0),a0,r)∗An=(f(n+1),an+1,r)
通过矩阵快速幂就可以推出
f(n)%1e9+7
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
using namespace std;
typedef long long LL;
const LL MOD = 1e9+7;
const int MAXN = 3;
struct Matrix{
LL mat[MAXN][MAXN];
};
Matrix Multi(Matrix a, Matrix b){
Matrix c;
for(int i=0; i<MAXN; i++){
for(int j=0; j<MAXN; j++){
c.mat[i][j] = 0;
for(int k=0; k<MAXN; k++){
c.mat[i][j] = c.mat[i][j] + a.mat[i][k]*b.mat[k][j] % MOD;
c.mat[i][j] = (c.mat[i][j] % MOD + MOD) % MOD;
}
}
}
return c;
}
Matrix Pow(Matrix a, LL n){
Matrix ans;
memset(ans.mat, 0, sizeof(ans.mat));
for(int i=0; i<MAXN; i++) ans.mat[i][i] = 1;
while(n){
if(n & 1) ans = Multi(ans, a);
n>>=1;
a = Multi(a, a);
}
return ans;
}
int main()
{
LL p, q, r, n;
while(~scanf("%lld%lld%lld%lld",&p,&q,&r,&n)){
Matrix ans;
ans.mat[0][0] = q, ans.mat[0][1] = 0, ans.mat[0][2] = 0;
ans.mat[1][0] = 3*p%MOD, ans.mat[1][1] = p, ans.mat[1][2] = 0;
ans.mat[2][0] = 3, ans.mat[2][1] = 1, ans.mat[2][2] = 1;
ans = Pow(ans, n);
LL ret = (ans.mat[2][0]*r%MOD+MOD)%MOD;
cout<<ret<<endl;
}
return 0;
}