[SCOI2016]围棋

Description
近日,谷歌研发的围棋AI—AlphaGo以4:1的比分战胜了曾经的世界冠军李世石,这是人工智能领域的又一里程碑。与传统的搜索式AI不同,AlphaGo使用了最近十分流行的卷积神经网络模型。在卷积神经网络模型中,棋盘上每一块特定大小的区域都被当做一个窗口。例如棋盘的大小为5×6,窗口大小为2×4,那么棋盘中共有12个窗口。此外,模型中预先设定了一些模板,模板的大小与窗口的大小是一样的。下图展现了一个5×6的棋盘和两个2×4的模板。对于一个模板,只要棋盘中有某个窗口与其完全匹配,我们称这个模板是被激活的,否则称这个模板没有被激活。例如图中第一个模板就是被激活的,而第二个模板就是没有被激活的。我们要研究的问题是:对于给定的模板,有多少个棋盘可以激活它。为了简化问题,我们抛开所有围棋的基本规则,只考虑一个n×m的棋盘,每个位置只能是黑子、白子或无子三种情况,换句话说,这样的棋盘共有3n×m种。此外,我们会给出q个2×c的模板。我们希望知道,对于每个模板,有多少种棋盘可以激活它。强调:模板一定是两行的。

Input
输入数据的第一行包含四个正整数n,m,c和q,分别表示棋盘的行数、列数、模板的列数和模板的数量。随后2×q行,每连续两行描述一个模板。其中,每行包含c个字符,字符一定是‘W’,‘B’或‘X’中的一个,表示白子、黑子或无子三种情况的一种。N<=100,M<=12,C<=6,Q<=5

Output
输出应包含q行,每行一个整数,表示符合要求的棋盘数量。由于答案可能很大,你只需要输出答案对1,000,000,007取模后的结果即可。

Sample Input
3 1 1 2
B
W
B
B

Sample Output
6
5


这题写的真是神清气爽……题解真是神仙写法

首先看范围,\(m\)辣么小肯定要状压

考虑使用补集转化,求没有任何一个子矩阵满足匹配条件的棋盘种数,然后我们暴力状压上一行状态,逐行转移,复杂度\(O(n*3^m+3^{2*m})\),直接TLE

改一下状压状态,因为我们只需要关系是否匹配,并不需要关心黑白或者无,所以我们可以用二进制状压

考虑使用轮廓线DP解决这个问题,设\(f_{i,j,S,x,y}\)表示当前考虑到第\(i\)行第\(j\)列,\(S\)记录轮廓线上每个点能否匹配完模板的第一行(\(S\)上第\(k\)位为1表示轮廓线上第\(k\)位,将模板第一行最后一个格子放置在此后,模板第一行颜色不会与已决策棋盘区域发生混乱),目前匹配到模板第一行的第\(x\)列,第二行的第\(y\)

然后前两维直接滚动掉,然后我们用KMP预处理出模板两行的失配函数,匹配的时候直接暴力枚举转移即可

每当枚举到新的一行时,要将上一行的值全部转移过来,具体实现可以看代码

/*program from Wolfycz*/
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define inf 0x7f7f7f7f
using namespace std;
typedef long long ll;
typedef unsigned int ui;
typedef unsigned long long ull;
inline char gc(){
	static char buf[1000000],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
inline int frd(){
	int x=0,f=1;char ch=gc();
	for (;ch<'0'||ch>'9';ch=gc())	if (ch=='-')    f=-1;
	for (;ch>='0'&&ch<='9';ch=gc())	x=(x<<1)+(x<<3)+ch-'0';
	return x*f;
}
inline int read(){
	int x=0,f=1;char ch=getchar();
	for (;ch<'0'||ch>'9';ch=getchar())	if (ch=='-')	f=-1;
	for (;ch>='0'&&ch<='9';ch=getchar())	x=(x<<1)+(x<<3)+ch-'0';
	return x*f;
}
inline void print(int x){
	if (x<0)    putchar('-'),x=-x;
	if (x>9)	print(x/10);
	putchar(x%10+'0');
}
const int N=(1<<12)*216,Mod=1e9+7,limit=235417;
struct S1{
	int S[N+10],x[N+10],y[N+10],v[N+10];
	int stack[N+10],top;
	void insert(int s,int _x,int _y,int val){
		ui res=s*limit*limit+_x*limit+_y; res%=N;
		if (!v[res])	stack[++top]=res,S[top]=s,x[top]=_x,y[top]=_y;
		v[res]=(v[res]+val)%Mod;
	}
	void clear(){while (top)	v[stack[top--]]=0;}
}f[2];
int v[2][10],Nxt[2][10],Fail[2][10][5];
char s[2][10];
int T(char x){return x=='B'?1:x=='W'?2:3;}
int mlt(int a,int b){
	int res=1;
	for (;b;b>>=1,a=1ll*a*a%Mod)	if (b&1)	res=1ll*res*a%Mod;
	return res;
}
int main(){
	int n=read(),m=read(),c=read(),q=read();
	while (q--){
		scanf("%s",s[0]+1);
		scanf("%s",s[1]+1);
		for (int i=1;i<=c;i++)	v[0][i]=T(s[0][i]);
		for (int i=1;i<=c;i++)	v[1][i]=T(s[1][i]);
		memset(Nxt,0,sizeof(Nxt));
		Nxt[0][0]=Nxt[1][0]=-1;
		for (int k=0;k<2;k++){
			for (int i=2,j=0;i<=c;i++){
				while (~j&&v[k][i]!=v[k][j+1])	j=Nxt[k][j];
				Nxt[k][i]=++j;
			}
		}
		for (int k=0;k<2;k++){
			for (int i=0;i<=c;i++){
				for (int j=i,x=1;x<=3;x++,j=i){
					while (~j&&x!=v[k][j+1])	j=Nxt[k][j];
					Fail[k][i][x]=++j;
				}
			}
		}
		int p=0; f[p].clear();
		f[p].insert(0,0,0,1);
		for (int i=1;i<=n;i++){
			for (int l=1;l<=f[p].top;l++)	f[p].x[l]=f[p].y[l]=0;
			for (int j=1;j<=m;j++){
				f[p^=1].clear();
				for (int l=1;l<=f[p^1].top;l++){
					int S=f[p^1].S[l],x=f[p^1].x[l],y=f[p^1].y[l];
					for (int k=1;k<=3;k++){
						int a=Fail[0][x][k],b=Fail[1][y][k];
						int tmp=(b==c)<<(j-1),sta=S;
						if (S&tmp)	continue;
						if (sta>>(j-1)&1)	sta^=1<<(j-1);
						if (a==c)	sta^=1<<(j-1);
						f[p].insert(sta,a,b,f[p^1].v[f[p^1].stack[l]]);
					}
				}
			}
		}
		int Ans=mlt(3,n*m);
		for (int i=1;i<=f[p].top;i++)	Ans=(Ans-f[p].v[f[p].stack[i]])%Mod;
		printf("%d\n",(Ans+Mod)%Mod);
	}
	return 0;
}
posted @ 2019-03-06 18:32  Wolfycz  阅读(320)  评论(0编辑  收藏  举报