51NOD 1341 混合序列(构造矩阵)

xiaoxiao2021-02-28  128

传送门 题目描述:

Input

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

Output

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

Input示例

2 2 1 1

Output示例

3

解题思路:

f(n)=ni=0aibni ,将其展开有: f(n)=a0bn+a1bn1+...+anb0 同理 f(n+1) 也可写出: f(n+1)=n+1i=0aibn+1i f(n+1)=a0bn+1+a1bn+...+an+1b0 通过观察,发现如下公式成立: f(n+1)=f(n)q+an+1b0 根据这个递推式,我们可以构造一个矩阵 A A=qb0pb00p1001 (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; }
转载请注明原文地址: https://www.6miu.com/read-38234.html

最新回复(0)