【BZOJ5505】[GXOI/GZOI2019]逼死强迫症（矩阵快速幂）

【BZOJ5505】[GXOI/GZOI2019]逼死强迫症（矩阵快速幂）

题面

BZOJ
洛谷

题解

如果没有那两个\(1*1\)的东西，答案就是斐波那契数，可以简单的用\(dp\)得到。
大概是设\(f[i]\)表示当前除了到第\(i\)列的方案数，转移是考虑用\(2*1\)竖着覆盖一列还是\(2\)个\(1*2\)横着覆盖两列，得到转移\(f[i]=f[i-1]+f[i-2]\)。
现在回假设要在这一行放上第二个\(1*1\)，那么直到前一个\(1*1\)所在列之前的所有方块都被唯一确定了，而左侧就是随便放的方案数，即斐波那契数列。
设\(g[i]\)表示上面的\(f[i]\)，\(f[i]\)为答案，\(s[i]\)为\(g[i]\)前缀和。
那么\(f[i]=f[i-1]+f[i-2]+2h[i-3]\)，即考虑当前在哪一行放上这个\(1*1\)的东西，有两种方案数，接下来可以在任何一个地方把这个东西补齐，而前面可以任意摆放。
根据斐波那契数列的性质，\(h[i-3]=g[i-1]-1\)，所以拿矩阵快速幂维护一下斐波那契数列和\(dp\)数列就行了。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define MOD 1000000007
inline int read()
{
	int x=0;bool t=false;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')t=true,ch=getchar();
	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
	return t?-x:x;
}
const int N=5;
struct Matrix
{
	int s[N][N];
	void clear(){memset(s,0,sizeof(s));}
	void init(){clear();for(int i=0;i<N;++i)s[i][i]=1;}
	int*operator[](int x){return s[x];}
}A[32],B,P,Ans;
Matrix operator*(Matrix a,Matrix b)
{
	Matrix c;c.clear();
	for(int i=0;i<N;++i)
		for(int j=0;j<N;++j)
			for(int k=0;k<N;++k)
				c[i][j]=(c[i][j]+1ll*a[i][k]*b[k][j])%MOD;
	return c;
}
Matrix fpow(int b)
{
	Matrix s;s.init();
	for(int i=0;b;++i,b>>=1)
		if(b&1)s=s*A[i];
	return s;
}
int main()
{
	B[0][2]=B[0][3]=B[0][4]=1;
	P[0][1]=P[1][0]=P[1][1]=P[2][3]=P[3][2]=P[3][3]=P[4][4]=1;
	P[2][1]=P[3][1]=2;P[4][1]=MOD-2;
	A[0]=P;for(int i=1;i<32;++i)A[i]=A[i-1]*A[i-1];
	int T=read();
	while(T--)
	{
		int n=read();
		if(n==1){puts("0");continue;}
		if(n==2){puts("0");continue;}
		Ans=B*fpow(n-2);
		printf("%d\n",Ans[0][1]);
	}
	return 0;
}