佳佳的斐波那契

佳佳的斐波那契

题目描述

---

核心思路

由题目可知，$S_n=f_1+f_2+\cdots +f_n$，$T_n=f_1+2f_2+\cdots +(n-1)f_{n-1}+n{f}_n$

我们设$p_n=n{S}_n-T_n$，则会有$p_n=(n-1)f_1+(n-2)f_2+\cdots +1f_{n-1}+0f_n$，记为【1】式

我们设$p_{n+1}=(n+1)S_{n+1}-T_{n+1}=n{f}_1+(n-1)f_2+\cdots +2f_{n-1}+1f_n$，记为【2】式

用【2】-【1】，可以得到：

$p_{n+1}-p_n=f_1+f_2+\cdots +f_{n-1}+f_n=S_n$，也就是有$p_{n+1}-p_n=S_n$

由于$p_n=n{S}_n-T_n$，因此如果我们想要求出$T_n$，则需要先求出$p_n$和$S_n$。

我们设$F_n$=$\left[\begin{array}{cccc}{f}_n& f_{n+1}& S_n& p_n\end{array}\right]$，设$F_{n+1}=\left[\begin{array}{cccc}{f}_{n+1}& f_{n+2}& S_{n+1}& p_{n+1}\end{array}\right]$，

构造一个矩阵$A=\left[\begin{array}{cccc}0& 1& 0& 0\\ 1& 1& 1& 0\\ 0& 0& 1& 1\\ 0& 0& 0& 1\end{array}\right]$，那么就会有$F_n\times A=F_{n+1}$。

通过递推可知，$F_n=F_1\times A^{n-1}$，其中$F_1$是边界，$F_1$=$\left[\begin{array}{cccc}{f}_1& f_2& S_1& p_1\end{array}\right]$，$f_1=1,f_2=1$，而$S_1=f_1=1$，由于$p_n=n{S}_n-T_n$，那么$p_1=S_1-T_1$，而$T_1=f_1=1$，所以$p_1=1-1=0$

所以，我们只需要用矩阵快速幂求出$A^{n-1}$就好了。然后$T_n=n{S}_n-p_n$，即可求得结果$T_n$。

---

代码

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=4;
typedef long long LL;
int n,m;
void mul(int a[][N],int b[][N],int c[][N])
{
    int tmp[N][N]={0};
    for(int i=0;i<N;i++)
    {
        for(int j=0;j<N;j++)
        {
            for(int k=0;k<N;k++)
            {
                tmp[i][j]=(tmp[i][j]+(LL)a[i][k]*b[k][j])%m;
            }
        }
    }
    memcpy(c,tmp,sizeof tmp);
}
int main()
{
    scanf("%d%d",&n,&m);
    //本来边界F1={1,1,1,0},但是我们这里为了只写一个mul函数,所以把它改造成了N*N的矩阵
    //这样就相当于只处理了两个矩阵相乘,方便了计算
    int F[N][N]={
            {1,1,1,0},
            {0,0,0,0},
            {0,0,0,0},
            {0,0,0,0}
    };
    //构造矩阵A
    int a[N][N]={
            {0,1,0,0},
            {1,1,1,0},
            {0,0,1,1},
            {0,0,0,1}
    };
    int b=n-1;
    while(b)
    {
        if(b&1)
            mul(F,a,F); //相当于res=res*a
        mul(a,a,a);     //相当于a=a*a
        b>>=1;
    }
    //注意这里有可能两个数相减结果会为负数,如果取模得到的结果就是负的
    //为了得到正的模数,可以采用(x%m+m)%m的方法
    printf("%d\n",(((LL)n * F[0][2] - F[0][3]) % m + m) % m);
}

---