20200612 SCOI模拟T2（状压dp+矩阵快速幂）

T2 P3977 [TJOI2015]棋盘

思路：
出题人不说那个模板是从零下标开始的……
（我求求他做个人吧

留意数据范围
发现对于一行的状态最多 64 种
于是考虑状压
可以暴力处理出一行有几种可行状态
以及可以向下转移的状态
有 dp 方程
$f[i][j]$ 表示第 i 行状态为 j 时的方案总数
k 为可以转移到 j 的状态
$$f[i][j]=\sum f[i-1][k]$$
暴力转移会 TLE，考虑优化
构造矩阵
$$\left[\begin{array}{ccc}{f}_1& f_2& f_3\\ 0& 0& 0\\ 0& 0& 0\end{array}\right]$$
为当前状态
乘上$总状态数\times 总状态数$的矩阵可以转移到下一状态
其中两两可以转移的状态的位置值为 1

代码：

#include<bits/stdc++.h>
using namespace std;
#define in Read()
typedef unsigned int uint;

inline char ch(){
	static char buf[1<<21],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++;
}

inline int in{
	int s=0,f=1;char x;
	for(x=getchar();x<'0'||x>'9';x=getchar())	if(x=='-')	f=-1;
	for( ;x>='0'&&x<='9';x=getchar())	s=(s*10)+(x&15);
	return f==1?s:-s;
}

const int A=100;
int n,m;
int P,K;
int a1[A],a2[A],num;
int p[A],tot;

struct node{
	uint p[A][A];
	node(){}
	inline friend node operator * (const node u,const node v){
		node ans;
		for(int i=1;i<=64;i++)
			for(int j=1;j<=64;j++)	ans.p[i][j]=0;
		for(int i=1;i<=tot;i++)
			for(int j=1;j<=tot;j++)
				for(int k=1;k<=tot;k++)
					ans.p[i][j]+=u.p[i][k]*v.p[k][j];
		return ans;
	}
}s,ans;

inline void scan(){
	n=in,m=in,P=in,K=in;
	for(int i=0;i<3;i++)
		for(int j=0;j<P;j++){
			int u=in;
			if(u==1&&(i!=1||j!=K)){
				a1[++num]=i-1;
				a2[num]=j-K;
			}
		}
	return;
}

inline int check(int x,int y){
	for(int i=0;i<m;i++){
		if(x&(1<<i)){
			for(int j=1;j<=num;j++){
				int now=i+a2[j];
				if(now>=m)	continue;
				if(a1[j]==0){
					if(x&(1<<now))	return 0;
				}
				else if(a1[j]==1){
					if(y&(1<<now))	return 0;
				}
			}
		}
		if(y&(1<<i)){
			for(int j=1;j<=num;j++){
				int now=i+a2[j];
				if(now>=m)	continue;
				if(a1[j]==0){
					if(y&(1<<now))	return 0;
				}
				else if(a1[j]==-1){
					if(x&(1<<now))	return 0;
				}
			}
		}
	}
	return 1;
}

inline void prepare(){
	for(int i=0;i<(1<<m);i++)
		if(check(i,0))	p[++tot]=i;
	for(int i=1;i<=tot;i++)
		for(int j=1;j<=tot;j++){
			if(check(p[i],p[j]))	s.p[i][j]=1;
		}
	ans.p[1][1]=1;
	return;
}

inline node quick(node f,node g,int c){
	while(c){
		if(c&1)	f=f*g;
		g=g*g;
		c>>=1;
	}
	return f;
} 

signed main(){
	scan();
	prepare();
	ans=quick(ans,s,n);
	int res=0;
	for(int i=1;i<=tot;i++)	
		res+=ans.p[1][i];
	printf("%u\n",res);
	return 0;
}