[CF 613 Div.1E] Puzzle Lover

这道题目……感觉细节爆炸级别的多啊……要不是去找人要了代码感觉都要调不出来了

题意

有一个\(2\times N\)的网格,其中每个网格上面都有一个字符,问上下左右走,能够走出多少个给定的字符串\(S\)?答案对\(10^9+7\)取模。
\(1\leq N \leq 2\times 10^3\)

分析

实际上可以发现,这样的路径肯定最多就是长这样的形式:左边一段绕回来的路径,中间一段可以上下跑得路径,右边一段绕回来的路径。

为什么是这样的呢?考虑两边“绕回来”的路径,因为每个节点只能被经过一次,所以必然不能在“绕回去”。那么这个分类正确性就很显然了。
左右两边,相当于是要上下两个长度相等的段合起来匹配,而且肯定是前缀或者后缀这样的,利用哈希即可。
进一步在中间做一个dp,\(f_{i,j,k}\)表示在\(i\)\(j\)列匹配到串的位置\(k\),把三段拼起来就好了(实际上是两段,我们考虑可以正着和反着各做一遍,每一次可以同时处理dp与一边的hash,全部算进dp里面,表示从某个点出去,匹配半条路径,也就是一段中间的+一段往回绕的,最后拼起来)。代码里面dp的细节比较多,非常难调……

# include <bits/stdc++.h>
# define re(i,a,b) for (int i=(a);i<(b);++i)
# define rep(i,a,b) for (int i=(a);i<=(b);++i)
# define cl(x) memset(x,0,sizeof(x))
using namespace std;
typedef long long ll;

inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read(char *s){
  char c=nc(); int len=0;
  for (;!(c>='a' && c<='z');c=nc());
  for (;c>='a' && c<='z';s[++len]=c,c=nc()); s[++len]=0; return len-1;
}
const int P=1e9+7;
const int N=2005;
inline void add(int &x,int y){
  x+=y; if (x>=P) x-=P; 
}
const ll S=31;
const ll MOD=987654321;
ll seed[N];
inline void Pre(int n){
  seed[0]=1;
  rep(i,1,n) seed[i]=seed[i-1]*S%MOD;
}
struct Hash{
  ll h[N];
  inline void make(int n,char *s){
    h[0]=0;
    rep(i,1,n) h[i]=(h[i-1]*S+s[i]-'a')%MOD;
  }
  inline ll cut(int l,int r){
    return (h[r]+MOD-h[l-1]*seed[r-l+1]%MOD)%MOD;
  }
}pre[2],suf[2],ss;

int n,m;
char a[2][N],s[N];
int f[2][N][N];

inline int Solve(int flag){
  int ret=0; cl(f);
  rep(j,1,n){
    f[0][j][0]=f[1][j][0]=1;
    re(i,0,2) rep(k,2,min(n-j+1,m/2))
    if (ss.cut(m-2*k+1,m-k)==pre[i].cut(j,j+k-1) && ss.cut(m-k+1,m)==suf[i^1].cut(n-(j+k-1)+1,n-j+1))
      if (2*k!=m || flag)
        add(ret,f[i][j][m-2*k]);
    re(i,0,2) rep(k,2,min(j,m/2))
    if (ss.cut(k+1,2*k)==pre[i].cut(j-k+1,j) && ss.cut(1,k)==suf[i^1].cut(n-j+1,n-(j-k+1)+1))
      if (2*k!=m || flag)
        add(f[i][j+1][2*k],1);
    re(i,0,2) re(k,0,m)
    if (a[i][j]==s[k+1]){
      add(f[i][j+1][k+1],f[i][j][k]);
      if (k+2<=m && a[i^1][j]==s[k+2])
        add(f[i^1][j+1][k+2],f[i][j][k]);
    }
    re(i,0,2)
      add(ret,f[i][j+1][m]);
  }
  return ret;
}

int main(){
  freopen("string.in","r",stdin);
  freopen("string.out","w",stdout);
  Pre(2000);
  n=read(a[0]); read(a[1]);
  re(i,0,2){
    pre[i].make(n,a[i]);
    reverse(a[i]+1,a[i]+n+1);
    suf[i].make(n,a[i]);
    reverse(a[i]+1,a[i]+n+1);
  }
  m=read(s); ss.make(m,s);
  int Ans=0;
  add(Ans,Solve(1));
  if (m>1){
    reverse(s+1,s+m+1);
    ss.make(m,s);
    add(Ans,Solve(0));
    if (m==2){
      rep(j,1,n) re(i,0,2)
      if (a[i][j]==s[1] && a[i^1][j]==s[2])
        add(Ans,P-1);
    }
  }
  printf("%d\n",Ans);
  return 0;
}
posted @ 2018-04-18 21:56  WenDavid  阅读(360)  评论(0编辑  收藏  举报