[牛客OI周赛3-提高组A] 地斗主 [dp][矩阵快速幂]

link

原题(poj)

这一类棋盘覆盖问题通常可以考虑“上一行覆盖到下一行的情况”然后进行转移。
这样的每次都有相同规律的递推可以考虑用矩阵加速。

官方题解：
随便手摸一下，我们可以得到…我们发现…我们可以把公式变为…………然后就推出矩阵了
我：？？？

关于矩阵里面对应的具体的状态可以见POJ3420

程序贴上来好像格式有点乱
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cctype>
#include<ctime>
#include<cstdlib>
using namespace std;
int T;
long long N,M,tmp[5][5]={};
long long transfer[5][5]={{1,1,1,1,0},{1,0,0,0,0},{2,0,1,0,0},{1,0,0,0,1},{0,0,0,1,0}};
long long qmpow(long long t)
{
	long long ans[5][5]={{1,0,0,0,0},{0,1,0,0,0},{0,0,1,0,0},{0,0,0,1,0},{0,0,0,0,1}};
	long long base[5][5],ret=0;
    for(int i=0;i<5;++i)
    {
    	for(int j=0;j<5;++j)
    	{
    		base[i][j]=transfer[i][j];
    	}
    }
    while(t)
	{
        if(t&1)
		{
			for(int i=0;i<5;++i)
			{
				for(int j=0;j<5;++j)
				{
					tmp[i][j]=0;
				}
			}
   			for(int i=0;i<5;++i)
			{
        		for(int j=0;j<5;++j)
				{
            		for(int k=0;k<5;++k)
					{
               			tmp[i][j]=(tmp[i][j]+ans[i][k]*base[k][j]%M)%M;
            		}
        		}
    		}
    		for(int i=0;i<5;++i)
    		{
    			for(int j=0;j<5;++j)
    			{
    				ans[i][j]=tmp[i][j];
    			}
    		}
        }
		for(int i=0;i<5;++i)
		{
			for(int j=0;j<5;++j)
			{
				tmp[i][j]=0;
			}
		}
  		for(int i=0;i<5;++i)
		{
       		for(int j=0;j<5;++j)
			{
        		for(int k=0;k<5;++k)
				{
           			tmp[i][j]=(tmp[i][j]+base[i][k]*base[k][j]%M)%M;
        		}
    		}
		}
		for(int i=0;i<5;++i)
		{
			for(int j=0;j<5;++j)
			{
				base[i][j]=tmp[i][j];
			}
 		}
        t>>=1;
    }
    ret=(ret+5ll*ans[0][0]%M)%M;
    ret=(ret+1ll*ans[1][0]%M)%M;
    ret=(ret+2ll*ans[2][0]%M)%M;
    ret=(ret+1ll*ans[3][0]%M)%M;
    ret=(ret+1ll*ans[4][0]%M)%M;
    return ret;
}
int main()
{
	scanf("%d",&T);
    while(T--)
	{
		scanf("%lld%lld",&N,&M);
        if(N==1ll)
		{
            printf("%lld\n",1ll%M);
            continue;
        }
        if(N==2ll)
		{
            printf("%lld\n",5ll%M);
            continue;
        }
        printf("%lld\n",qmpow(N-2ll));
    }
    return 0;
}