矩阵快速幂 - 佳佳的斐波那契 - AcWing 1304

矩阵快速幂 - 佳佳的斐波那契 - AcWing 1304

佳佳对数学，尤其对数列十分感兴趣。

在研究完 Fibonacci 数列后，他创造出许多稀奇古怪的数列。

例如用 S(n) 表示 Fibonacci 前 n 项和 mod m 的值，

$即S(n)=(F_1+F_2+\dots +F_n)modm，其中F_1=F_2=1,F_i=F_{i-1}+F_{i-2}。$

可这对佳佳来说还是小菜一碟。

终于，她找到了一个自己解决不了的问题。

$用T(n)=(F_1+2F_2+3F_3+\dots +n{F}_n)modm表示Fibonacci数列前n项变形后的和modm的值。$

现在佳佳告诉你了一个 n 和 m，请求出 T(n) 的值。

输入格式

共一行，包含两个整数 n 和 m。

输出格式

共一行，输出 T(n) 的值。

数据范围

$1\le n,m\le 2^{31}-1$

输入样例：

5 5

输出样例：

1

样例解释

$T(5)=(1+2\times 1+3\times 2+4\times 3+5\times 5)mod5=1$

---

分析：

$求S_n：$矩阵快速幂 - 斐波那契前 n 项和 - AcWing 1303

$已知S_n与F_n的关系，现要将T_n与S_n和F_n建立关系，通过已知来求未知，$

$观察发现，T_n相比于S_n，关于n的次数要高一次，$

$我们先给S_n升幂，乘一个n，$

$\{\begin{array}{l}{T}_n=F_1+2F_2+3F_3+\dots +n{F}_n\\ \\ nS(n)=n{F}_1+n{F}_2+n{F}_3+\dots +n{F}_n\end{array}$

$记P_n=n{S}_n-T_n=(n-1)F_1+(n-2)F_2+(n-3)F_3+...+F_{n-1}①$

$则P_{n+1}=(n+1)S_{n+1}-T_{n+1}=n{F}_1+(n-1)F_2+(n-2)F_3+...+F_n②$

$②-①得P_{n+1}-P_n=F_1+F_2+...+F_n=S_n，即P_{n+1}=P_n+S_n$

$到这里，我们就将S_n、P_n、F_n之间建立了递推关系，求出P_n后，可以解得：T_n=n{S}_n-P_n。$

$设矩阵Q_n=[F_n{F}_{n+1}{S}_n{P}_n]，现需凑出4\times 4变换矩阵A，使得Q_{n}A=Q_{n+1}$

$F_{n+1}=0\times F_n+1\times F_{n+1}+0\times S_n+0\times P_n$

$F_{n+2}=1\times F_n+1\times F_{n+1}+0\times S_n+0\times P_n$

$S_{n+1}=0\times F_n+1\times F_{n+1}+1\times S_n+0\times P_n$

$P_{n+1}=0\times F_n+0\times F_{n+1}+1\times S_n+1\times P_n$

$于是变换矩阵A=\left|\begin{array}{c}0100\\ 1110\\ 0011\\ 0001\end{array}\right|，初始矩阵Q_1=[F_1{F}_2{S}_1{P}_1]=[1110]$

$先计算A^{n-1}，再计算Q_n=Q_1{A}^{n-1}=[F_n{F}_{n+1}{S}_n{P}_n]$

$最后得到T_n=n{S}_n-P_n，注意负数取模。$

代码：

#include<iostream>
#include<cstring>
#include<algorithm>

#define ll long long

using namespace std;

const int N=4;

int n,m;
ll f1[N][N]={{1,1,1,0}};  //Pn=n*Sn-Tn
ll A[N][N]={
    {0,1,0,0},
    {1,1,1,0},
    {0,0,1,1},
    {0,0,0,1}
};

void mul(ll a[][N],ll b[][N])
{
    ll c[N][N]={0};
    for(int i=0;i<N;i++)
        for(int j=0;j<N;j++)
            for(int k=0;k<N;k++)
                c[i][j]=(c[i][j]+a[i][k]*b[k][j]%m)%m;
    memcpy(a,c,sizeof c);
}

void quick_pow(ll a[][N],int k)
{
    ll E[N][N]={
        {1,0,0,0},
        {0,1,0,0},
        {0,0,1,0},
        {0,0,0,1}
    };
    while(k)
    {
        if(k&1) mul(E,a);
        mul(a,a);
        k>>=1;
    }
    memcpy(a,E,sizeof E);
}

int main()
{
    cin>>n>>m;
    
    quick_pow(A,n-1);
    mul(f1,A);
    
    ll ans=(n*f1[0][2]%m-f1[0][3]+m)%m;
    cout<<ans<<endl;
    
    return 0;
}