斐波那契数列卷积
示例1
输入
3
输出
2
示例2
输入
19260817
输出
511682927
解题思路
提前我们要知道 Fn=Fn-1+Fn-2
规律是适合的,因为写起来太麻烦所以这就举个例子说明吧,
A7= F0* F7 + F1* F6 + F2* F5 + F3* F4 + F4* F3 + F5* F2 + F6* F1 + F7* F0
A6= F0* F6 + F1* F5 + F2* F4 + F3* F3 + F4* F2 + F5* F1 + F6* F0
A7-A6=F0*(F7-F6) + F1*(F6-F5) + F2*(F5-F4) + F3*(F4-F3) + F4*(F3-F2) + F5*(F2-F1) + F6*(F1-F0) + F7* F0
因为 F1=1,F0=0;所以F6*(F1-F0)=F6;
A7-A6=F0* F5 + F1* F4 + F2* F3 + F3* F2 + F4* F1 + F5* F0 + F6
A7-A6=A5+F6;
也就是 An=An-1+An-2+Fn-1
这样就可以用矩阵快速幂来构造一个矩阵来求了(感谢bly提供的方程,tql)
代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 998244353;
struct node{
ll a[4][4];
};
node mul(node a,node b)
{
node ans;
for(int i=0;i<4;i++)
for(int j=0;j<4;j++)
ans.a[i][j]=0;
for(int i=0;i<4;i++)
{
for(int j=0;j<4;j++)
{
for(int k=0;k<4;k++)
{
ans.a[i][j]=(ans.a[i][j]+((a.a[i][k]%mod)*(b.a[k][j]%mod))%mod)%mod;
}
}
}
return ans;
}
node poww(node a,ll b)
{
node ans;
memset(ans.a,0,sizeof(ans.a));
for(int i=0;i<4;i++) ans.a[i][i]=1;
while(b)
{
if(b&1) ans=mul(ans,a);
a=mul(a,a);
b>>=1;
}
return ans;
}
node yl,bly;
int main()
{
ios::sync_with_stdio(0);
ll n;
cin>>n;
memset(yl.a,0,sizeof(yl.a));
memset(bly.a,0,sizeof(bly.a));
yl.a[0][0]=1;
yl.a[0][1]=1;
yl.a[0][2]=1;
yl.a[1][0]=1;
yl.a[2][2]=1;
yl.a[2][3]=1;
yl.a[3][2]=1;
bly.a[2][0]=1;
bly.a[0][0]=1;
bly.a[3][0]=1;
if(n==1) cout<<0<<endl;
else if(n==2) cout<<1<<endl;
else
{
node ans=poww(yl,n-2);
ans=mul(ans,bly);
cout<<ans.a[0][0]<<endl;
}
return 0;
}