1. 程式人生 > >[GXOI/GZOI2019]逼死強迫癥

[GXOI/GZOI2019]逼死強迫癥

代碼 main time char std rebuild ons isp ref

題目

設我們最後的答案是\(g_n\)

我們發現在最後豎著放一個\(2\times 1\)的,和橫著放兩個\(1\times 2\)的就可以區分開之前的方案了

所以如果僅僅使用\(1\times 2\)的塊來填滿\(2\times n\)的格子,方案數就是\(fib_n\)

於是

\[g_n=g_{n-1}+g_{n-2}+2\sum_{i=3}fib_{n-i}\]

後面就是\(fib\)數列,就是把那兩個\(1\times 1\)的在最後\(i\)列裏強行區分開

於是我們矩乘就好了

代碼

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define re register
#define LL long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
inline int read() {
    char c=getchar();int x=0;while(c<'0'||x>'9') c=getchar();
    while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-48,c=getchar();return x;
}
const int mod=1e9+7;
struct mat{int a[7][7];}a,s;
inline void reBuild() {
    for(re int i=0;i<7;i++)
        for(re int j=0;j<7;j++)
            a.a[i][j]=0;
    a.a[0][1]=a.a[1][0]=a.a[1][1]=1;
    a.a[2][3]=a.a[3][3]=a.a[3][2]=1;
    a.a[4][5]=a.a[5][6]=1;a.a[1][4]=2;
    a.a[6][3]=a.a[6][2]=a.a[6][6]=1;
}
inline mat operator*(mat a,mat b) {
    mat c;
    for(re int i=0;i<7;i++)
        for(re int j=0;j<7;j++)
            c.a[i][j]=0;    
    for(re int k=0;k<7;k++)
        for(re int i=0;i<7;i++)
            for(re int j=0;j<7;j++)
                c.a[i][j]=(c.a[i][j]+1ll*a.a[i][k]*b.a[k][j]%mod)%mod;
    return c;
}
inline void ksm(int b) {
    s=a;
    while(b) {if(b&1) s=s*a;b>>=1;a=a*a;}
}
int main() {
    int T=read(),x;
    while(T--) {
        reBuild();x=read();
        ksm(x);printf("%d\n",s.a[1][2]);
    }
    return 0;
}

[GXOI/GZOI2019]逼死強迫癥