【SNOI2019】纸牌（麻将DP）（矩阵快速幂）

传送门

---

今年省选好多麻将DP。。。

题解：

说的是纸牌，但是写过麻将DP的都知道这是麻将DP的一个套路。

以下把三张连着的牌叫做顺子，三张相同的牌叫做刻子。

不允许同类型顺子出现超过两次，否则我们可以用三个刻子来换掉。

实际上我们发现影响我们用多少张$i$牌的只有前面的$i-2,i-1$开头的顺子有多少个。

决定方案数不同的实际上是我们在当前状态下有多少空闲的刻子可以放。

把$i-2,i-1$的顺子个数压成一个状态，发现转移可以写成矩阵的形式，注意到只有题目给出的地方需要特殊处理，中间直接用快速幂转移即可。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

using std::cerr;
using std::cout;

cs int mod=998244353;
inline int add(int a,int b){a+=b-mod;return a+(a>>31&mod);}
inline int dec(int a,int b){a-=b;return a+(a>>31&mod);}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline void Inc(int &a,int b){a+=b-mod;a+=a>>31&mod;}
inline void Dec(int &a,int b){a-=b;a+=a>>31&mod;}
inline void Mul(int &a,int b){a=mul(a,b);}

struct mat{
	int a[9][9];
	mat(){memset(a,0,sizeof a);}
	int *operator[](int o){return a[o];}
	cs int *operator[](int o)cs{return a[o];}
	friend mat operator*(cs mat &A,cs mat &B){
		mat C;
		for(int re i=0;i<9;++i)
		for(int re j=0;j<9;++j)if(int t=A[i][j])
		for(int re k=0;k<9;++k)Inc(C[i][k],mul(t,B[j][k]));
		return C;
	}
};

mat A,B[61],tmp;
ll n,k[1007];
int C,X,a[1007];

inline mat pw_B(ll t){
	mat res;
	for(int re i=0;i<9;++i)res[i][i]=1;
	for(int re i=0;i<61;++i)if(t&(1ll<<i))res=res*B[i];
	return res;
}

signed main(){
#ifdef zxyoi
	freopen("card.in","r",stdin);
#endif
	scanf("%lld%d%d",&n,&C,&X);
	for(int re i=1;i<=X;++i)scanf("%lld%d",k+i,a+i);
	for(int re i=0;i<3;++i)
	for(int re j=0;j<3;++j)
	for(int re k=0;k<3;++k)
	if(i+j+k<=C)B[0][j*3+k][i*3+j]=(C-i-j-k)/3+1;
	for(int re i=1;i<61;++i)B[i]=B[i-1]*B[i-1];
	A[0][0]=1;
	for(int re i=1;i<=X;++i){
		A=A*pw_B(k[i]-k[i-1]-1);
		memset(tmp.a,0,sizeof tmp.a);
		for(int re j=0;j<3;++j)
		for(int re k=0;k<3;++k)
		for(int re l=0;l<3;++l){
			int use=j+k+l,nd=(use>a[i])?use:(a[i]+((use-a[i])%3+3)%3);
			if(nd<=C)tmp[j*3+k][l*3+j]=(C-nd)/3+1;
		}
		A=A*tmp;
	}A=A*pw_B(n-k[X]);
	cout<<A[0][0]<<"\n";
	return 0;
}