题解 19CSP-S十一集训三地联考—众神归位 A. 【十一·联考】幸福

$$幸福$$

Time Limit: 1S		Memory Limit: 256MB
Total Submissions: 652		Accepted: 111

$Description$
在只有纯白的世界里，栋栋和糯糯开始构建世界的色彩。

在第 $0$ 天，他们相爱并创造了一颗爱心，这颗爱心由栋栋拿着。

接下来的每一天，栋栋和糯糯都将手里的爱心全部递给对方，然后栋栋会再从手中再生出与其今天给出的相同数量的爱心。

即，若令第 ii 天的栋栋手里爱心的数目为 $f_i$，那么 $f$ 满足 $f_0=1,f_1=1,f_i=f_{i-1}+f_{i-2}(i>1)$ 。

不断再生并相互传递爱心会产生幸福，栋栋的幸福度被认为是其每天手里的爱心数的一个卷积，即 $F_n={\sum }_{i=0}^n{f}_i{f}_{n-i}$ 。

本是一成不变的风景中，两人逐渐被越来越多的爱心围绕。糯糯想要知道，从第 $0$ 天到第 $n$天栋栋一共收获了多少幸福呢？即要求你计算 ${\sum }_{i=0}^n{F}_i$ 。由于栋栋肯定很幸福，答案也可能很大，她只需要你将答案对 $998244353$ 取模。

$Input$
一行，一个整数 $n$ 。

$Output$
一行，一个整数，表示第 $0$ 天到第 $n$ 天的栋栋的幸福度的和。

$SampleInput$
5

$SampleOutput$
76

---

推一下公式：
$F_n={\sum }_{i=0}^n{f}_i{f}_{n-i}$
$=f_0{f}_n+f_1{f}_{n-1}+{\sum }_{i=2}^n{f}_i{f}_{n-i}$
$=f_0{f}_n+f_1{f}_{n-1}+{\sum }_{i=2}^n(f_{i-1}+f_{i-2})f_{n-i}$
$=f_0{f}_n+f_1{f}_{n-1}+{\sum }_{i=2}^n(f_{i-1}{f}_{n-i}+f_{i-2}{f}_{n-i})$
$=f_0{f}_n+f_1{f}_{n-1}+{\sum }_{i=2}^n{f}_{i-1}{f}_{n-i}+{\sum }_{i=2}^n{f}_{i-2}{f}_{n-i}$
$=f_0{f}_n+f_1{f}_{n-1}+{\sum }_{i=1}^n{f}_i{f}_{n-i-1}+{\sum }_{i=0}^n{f}_i{f}_{n-i-2}$
$=f_0{f}_n+f_1{f}_{n-1}+{\sum }_{i=0}^n{f}_i{f}_{n-i-1}-f_0{f}_{n-i}+{\sum }_{i=2}^n{f}_{i-2}{f}_{n-i}$
$=f_0{f}_n+{\sum }_{i=0}^n{f}_i{f}_{n-i-1}+{\sum }_{i=2}^n{f}_{i-2}{f}_{n-i}$
$=F_{n-1}+F_{n-2}+f_n$

然后用矩阵快速幂实现即可：
$$\left(\begin{array}{c}{f}_n\\ f_{n+1}\\ F_n\\ F_{n+1}\\ {\sum }_{i=0}^{n}F\end{array}\right)=\left(\begin{array}{ccccc}0& 1& 0& 0& 0\\ 1& 1& 0& 0& 0\\ 0& 0& 0& 1& 0\\ 1& 1& 1& 1& 0\\ 0& 0& 0& 1& 1\end{array}\right)\times \left(\begin{array}{c}{f}_{n-1}\\ f_n\\ F_{n-1}\\ F_n\\ {\sum }_{i=0}^{n-1}F\end{array}\right)={\left(\begin{array}{ccccc}0& 1& 0& 0& 0\\ 1& 1& 0& 0& 0\\ 0& 0& 0& 1& 0\\ 1& 1& 1& 1& 0\\ 0& 0& 0& 1& 1\end{array}\right)}^n\times \left(\begin{array}{c}1\\ 1\\ 1\\ 2\\ 1\end{array}\right)$$
代码实现如下：

#include <bits/stdc++.h>
#define ll long long
using namespace std;

const int mod=998244353;

struct maxtri
{
	ll a[10][10];
	maxtri() {memset(a,0,sizeof(a));}
	void build()
	{
		for(int i=1;i<=5;++i)
			a[i][i]=1;
	}
};

maxtri operator *(const maxtri &x,const maxtri &y)
{
	maxtri res;
	for(int k=1;k<=5;++k)
		for(int i=1;i<=5;++i) if(x.a[i][k]!=0)
			for(int j=1;j<=5;++j)
				res.a[i][j]=(res.a[i][j]+x.a[i][k]*y.a[k][j]%mod)%mod;
	return res;
}

maxtri fpow(maxtri a,ll k)
{
	maxtri res; res.build();
	while(k)
	{
		if(k&1) res=res*a;
		a=a*a;
		k>>=1;
	}
	return res;
}

int main()
{
//	freopen("input.txt","r",stdin);
	ll n; scanf("%lld",&n);
    maxtri ans,base;
    ans.a[1][1]=1; ans.a[2][1]=1; ans.a[3][1]=1; ans.a[4][1]=2; ans.a[5][1]=1;
    base.a[1][2]=base.a[2][1]=base.a[2][2]=base.a[3][4]=base.a[4][1]=base.a[4][2]=base.a[4][3]=base.a[4][4]=base.a[5][4]=base.a[5][5]=1;
    ans=fpow(base,n)*ans;
    printf("%lld",ans.a[5][1]);
    return 0;
}