【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;
}