【JZOJ4798】天使的分裂

Description

Solution

我们把$F_n$化简。

$F_n=\sum_{i=0}^nf_if_{n-i}=\sum_{i=2}^n(f_{i-1}+f_{i-2})f_{n-i}+f_0f_n+f_1f_{n-1}$

拆开得：
$\sum_{i=2}^nf_{i-1}f_{n-i}+\sum_{i=2}^nf_{i-2}f_{n-i}+f_0f_n+f_1f_{n-1}$

现在我们要将求和的部分用$F$来表示。

我们设$n_1=n-1，n_2=n-2$

对于第一部分：$\sum_{i=2}^nf_{i-1}f_{n-i}$，我们把$i$从$1$开始，那么原式变成：$\sum_{i=1}^{n_1}f_{i}f_{n_1-i}$。

然后我们逐步靠近$F$的式子，也就是把$i$变成$0$，原式就变成了：
$\sum_{i=0}^{n_1}f_{i}f_{n_1-i}-f_0f_{n_1}=F_{n1}-f_0f_{n_1}=F_{n-1}-f_0f_{n-1}$。

然后处理第二部分：$\sum_{i=2}^nf_{i-2}f_{n-i}$=$\sum_{i=0}^{n_2}f_{i}f_{n_2-i}=F_{n_2}=F_{n-2}$

然后整理一下可得：$F_{n-1}+F_{n-2}-f_0f_{n-1}+f_0f_n+f_1f_{n-1}$

由于$f_0=f_1=1$，所以化简可得$F_{n-1}+F_{n-2}-f_n-1+f_{n+1}=F_{n-1}+F_{n-2}+f_n$

我们发现递推式出来了。可这样的时间复杂度是$O(n)$的。

不虚，我们有矩阵乘法。

Code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define mo 998244353
#define ll long long
#define N 7
using namespace std;
ll a[N]={0,1,1,2,1,3,1};
ll z[N][N]=
{
{0},
{0,1,1,1,0,1},
{0,1,0,1,0,1},
{0,0,0,1,1,1},
{0,0,0,1,0,1},
{0,0,0,0,0,1},
};
ll b[N][N],c[N][N];
void mul(ll p[N][N],ll q[N][N])
{
    memset(c,0,sizeof(c));
    fo(i,1,N-1)
    fo(j,1,N-1)
    fo(k,1,N-1)
    c[i][j]=(c[i][j]+p[i][k]*q[k][j]%mo)%mo;
    memcpy(p,c,sizeof(c)); 
}
void pow(ll n)
{
    fo(i,1,N-1) b[i][i]=1;
    while(n)
    {
        if(n%2) mul(b,z);
        n/=2;
        mul(z,z);
    }
}
ll as[N];
int main()
{
    ll n;
    cin>>n;
    if(n<2)
    {
        cout<<a[n+4];
        return 0;
    }
    pow(n-1);
    fo(j,1,N-1)
    fo(k,1,N-1)
    as[j]=(as[j]+a[k]*b[k][j]%mo)%mo;
    cout<<as[5];
}