HDU4758 Walk Through Squares(AC自动机+状压DP)

题目大概说有个n×m的格子,有两种走法,每种走法都是一个包含D或R的序列,D表示向下走R表示向右走。问从左上角走到右下角的走法有多少种走法包含那两种走法。

D要走n次,R要走m次,容易想到用AC自动机上的DP解决:

  • 用两种走法的序列构造AC自动机
  • dp[i][j][S][k]表示D用了i个R用了j个,且当前走到自动机的S结点,已经包含的走法的状态集合是k的方案数
  • 转移就是走R或者走D了,我用我为人人来转移
 #include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
#define MAXN 222
int tn,ch[MAXN][],fail[MAXN],flag[MAXN];
int idx[];
void insert(char *s,int k){
int x=;
for(int i=; s[i]; ++i){
int y=idx[s[i]];
if(ch[x][y]==) ch[x][y]=++tn;
x=ch[x][y];
}
flag[x]|=(<<k);
}
void getfail(){
memset(fail,,sizeof(fail));
queue<int> que;
for(int i=; i<; ++i){
if(ch[][i]) que.push(ch[][i]);
}
while(!que.empty()){
int x=que.front(); que.pop();
for(int i=; i<; ++i){
if(ch[x][i]){
que.push(ch[x][i]);
fail[ch[x][i]]=ch[fail[x]][i];
flag[ch[x][i]]|=flag[ch[fail[x]][i]];
}else ch[x][i]=ch[fail[x]][i];
}
}
} int d[][][MAXN][]; int main(){
idx['R']=; idx['D']=;
int t,n,m;
char str[];
scanf("%d",&t);
while(t--){
tn=;
memset(ch,,sizeof(ch));
memset(flag,,sizeof(flag));
scanf("%d%d",&n,&m);
for(int i=; i<; ++i){
scanf("%s",str);
insert(str,i);
}
getfail();
memset(d,,sizeof(d));
d[][][][]=;
for(int len=; len<n+m; ++len){
for(int i=; i<=len; ++i){
int j=len-i;
if(j>m || i>n) continue;
for(int s=; s<=tn; ++s){
for(int k=; k<; ++k){
d[i+][j][ch[s][]][k|flag[ch[s][]]]+=d[i][j][s][k];
d[i+][j][ch[s][]][k|flag[ch[s][]]]%=;
d[i][j+][ch[s][]][k|flag[ch[s][]]]+=d[i][j][s][k];
d[i][j+][ch[s][]][k|flag[ch[s][]]]%=;
}
}
}
}
int res=;
for(int i=; i<=tn; ++i){
res+=d[n][m][i][];
res%=;
}
printf("%d\n",res);
}
return ;
}
上一篇:如何加密android apk


下一篇:Django 1.10 中文文档------3.2.1 模型Models