【bzoj4000】[TJOI2015]棋盘 dp+矩乘

题目坑爹，行列是从0开始算的，所以样例中是指最中间那个1为棋子

f[i][S]表示考虑到第i行，第i行状态为S的方案数

矩乘优化即可

开内存小了1，蛋疼。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#define ll unsigned int

using namespace std;

struct yts
{
int x,y;
ll a[(1<<6)+1][(1<<6)+1];
}a,b,c,ans;

int A[10][10],B[3][10];
int n,m,p,k,size;

void mul(yts &a,yts &b,yts &c)
{
memset(ans.a,0,sizeof(ans.a));
for (int i=1;i<=a.x;i++)
for (int j=1;j<=b.y;j++)
for (int k=1;k<=a.y;k++)
ans.a[i][j]+=a.a[i][k]*b.a[k][j];
c.x=a.x;c.y=b.y;
for (int i=1;i<=c.x;i++)
for (int j=1;j<=c.y;j++)
c.a[i][j]=ans.a[i][j];
}

void power(yts &a,int y)
{
c.x=size;c.y=size;
memset(c.a,0,sizeof(c.a));
for (int i=1;i<=size;i++) c.a[i][i]=1;
while (y)
{
if (y&1) mul(c,a,c);
mul(a,a,a);
y>>=1;
}
}

bool check(int x,int y)
{
memset(B,0,sizeof(B));
for (int i=1;i<=m;i++) if (x&(1<<(i-1))) B[1][i]=1;
for (int i=1;i<=m;i++) if (y&(1<<(i-1))) B[2][i]=1;
for (int i=1;i<=2;i++)
for (int j=1;j<=m;j++)
if (B[i][j])
for (int k=1;k<=A[2][0];k++)
{
if (j+A[2][k]<1 || j+A[2][k]>m) continue;
if (B[i][j+A[2][k]]) return 0;
}
for (int j=1;j<=m;j++)
if (B[1][j])
for (int k=1;k<=A[3][0];k++)
{
if (j+A[3][k]<1 || j+A[3][k]>m) continue;
if (B[2][j+A[3][k]]) return 0;
}
for (int j=1;j<=m;j++)
if (B[2][j])
for (int k=1;k<=A[1][0];k++)
{
if (j+A[1][k]<1 || j+A[1][k]>m) continue;
if (B[1][j+A[1][k]]) return 0;
}
return 1;
}

int main()
{
scanf("%d%d",&n,&m);
scanf("%d%d",&p,&k);
k++;
for (int i=1;i<=3;i++)
for (int j=1;j<=p;j++)
{
int x;
scanf("%d",&x);
if (x && (i!=2 || j!=k)) A[i][++A[i][0]]=j-k;
}
size=1<<m;
a.x=size;a.y=size;
for (int i=0;i<size;i++)
for (int j=0;j<size;j++)
a.a[i+1][j+1]=check(i,j);
power(a,n);
b.x=1;b.y=size;
b.a[1][1]=1;
mul(b,c,b);
ll ans=0;
for (int i=1;i<=size;i++) ans+=b.a[1][i];
printf("%u\n",ans);
return 0;
}