【JZOJ 4798】 天使的分裂

Description

Analysis

60分可以预处理斐波那契数列然后乱搞
100分是有递推式的==
$F_n=F_{n-1}+F_{n-2}+f_i$
至于推理过程，我没有看懂QAQ
我只知道这个如何证明
这里给出一种chunkit_lau的证明方法：
想象排成一列的n个点，如果两个点相乘则给这两个点连线
显然连线满足乘法分配律，所以连线可以互相转化
易证
有了这个就能直接矩阵乘法算了

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int N=5,mo=998244353;
typedef long long ll;
typedef ll matrix[N][N];
matrix A={
{0,1,0,0,1},
{1,1,0,0,1},
{0,1,0,1,1},
{0,1,1,1,1},
{0,0,0,0,1}};
matrix a,c;
ll n;
void mul(matrix a,matrix b)
{
    memset(c,0,sizeof(c));
    fo(i,0,4)
        fo(j,0,4)
            fo(k,0,4) (c[i][j]+=a[i][k]*b[k][j]%mo)%=mo;
}
void qmi(ll n)
{
    for(;n;n>>=1)
    {
        if(n&1)
        {
            mul(a,A);
            memcpy(a,c,sizeof(c));
        }
        mul(A,A);
        memcpy(A,c,sizeof(c));
    }
}
int main()
{
    scanf("%lld",&n);
    if(n==0)
    {
        printf("1");
        return 0;
    }
    a[0][0]=1,a[0][1]=2,a[0][2]=1,a[0][3]=1,a[0][4]=3;
    qmi(n-1);
    printf("%lld",a[0][4]);
    return 0;
}