51NOD 1341 混合序列（构造矩阵）

传送门
题目描述：

Input

共1行，4个整数数p, q, r, n中间用空格分隔(1 <= p, q, r, n<=1000000000)。

Output

对于每一个数据，在一行中输出答案。

Input示例

2 2 1 1

Output示例

3

解题思路：

令 $f(n)=\sum_{i=0}^na_i*b_{n-i}$，将其展开有：
$f(n)=a_0*b_n+a_1*b_{n-1}+...+a_n*b_0$
同理 $f(n+1)$ 也可写出：
$f(n+1)=\sum_{i=0}^{n+1}a_i*b_{n+1-i}$
$f(n+1)=a_0*b_{n+1}+a_1*b_{n}+...+a_{n+1}*b_0$
通过观察，发现如下公式成立：
$f(n+1)=f(n)*q+a_{n+1}*b_0$
根据这个递推式，我们可以构造一个矩阵 $A$：

$$A= \begin{pmatrix} q & 0 & 0 \\ b_0*p & p & 0 \\ b_0 & 1 & 1 \\ \end{pmatrix}$$
$(f(n),a_n,r)*A=(f(n+1),a_{n+1},r)$，即有：

$(f(0),a_0,r)*A^n=(f(n+1),a_{n+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;
}