【Codeforces 613E】—Puzzle Lover(DP)

传送门

考虑最后路径一定是分三段
左边的先向左走再走回来
中间上下晃
右边先向右走再拐回来

考虑把前2部分合在一起算
每次就把当前列新的可行的第一种加入dpdp数组即可
枚举右边用哈希算一下答案

为了方便就把SS反过来
不过要判一些重复情况

具体看代码(太麻烦了)

#include<bits/stdc++.h>
using namespace std;
const int RLEN=1<<20|1;
inline char gc(){
    static char ibuf[RLEN],*ib,*ob;
    (ob==ib)&&(ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
    return (ob==ib)?EOF:*ib++;
}
#define gc getchar
inline int read(){
    char ch=gc();
    int res=0,f=1;
    while(!isdigit(ch))f^=ch=='-',ch=gc();
    while(isdigit(ch))res=(res+(res<<2)<<1)+(ch^48),ch=gc();
    return f?res:-res;
}
#define ll long long
#define re register
#define pii pair<int,int>
#define pic pair<int,char>
#define fi first
#define se second
#define pb push_back
#define cs const
#define bg begin
#define poly vector<int>
#define chemx(a,b) ((a)<(b)?(a)=(b):0)
#define chemn(a,b) ((a)>(b)?(a)=(b):0)
cs int mod=1e9+7;
inline int add(int a,int b){return (a+=b)>=mod?a-mod:a;}
inline void Add(int &a,int b){(a+=b)>=mod?a-=mod:0;}
inline int dec(int a,int b){return (a-=b)<0?a+mod:a;}
inline void Dec(int &a,int b){(a-=b)<0?a+=mod:0;}
inline int mul(int a,int b){return 1ll*a*b%mod;}
inline void Mul(int &a,int b){a=1ll*a*b%mod;}
inline int ksm(int a,int b,int res=1){for(;b;b>>=1,Mul(a,a))(b&1)&&(Mul(res,a),1);return res;}
inline int Inv(int x){return ksm(x,mod-2);}
cs int N=2005;
int n,m;
char a[2][N],s[N];
int f[2][N][N];
int pw1[N],pw2[N];
cs int bas1=1331,mod1=1920643713,bas2=233,mod2=771939933;
inline void init(){
	pw1[0]=pw2[0]=1;
	for(int i=1;i<N;i++)pw1[i]=1ll*pw1[i-1]*bas1%mod1,pw2[i]=1ll*pw2[i-1]*bas2%mod2;
}
struct Hash{
	int has1[N],has2[N];
	inline void init(char *s){
		for(int i=1,len=strlen(s+1);i<=len;i++)
		has1[i]=(1ll*has1[i-1]*bas1+s[i]-'a'+1)%mod1,has2[i]=(1ll*has2[i-1]*bas2+s[i]-'a'+1)%mod2;
	}
	inline pii query(int l,int r){
		return pii(((has1[r]-1ll*has1[l-1]*pw1[r-l+1])%mod1+mod1)%mod1,((has2[r]-1ll*has2[l-1]*pw2[r-l+1])%mod2+mod2)%mod2);
	}
}S,pre[2],suf[2];
inline bool operator ==(cs pii &a,cs pii &b){
	return a.fi==b.fi&&a.se==b.se;
}
inline int calc(char *s,int dir){
	memset(f,0,sizeof(f));
	int res=0;
	for(int i=0;i<=n;i++){
		for(int k=2;k*2<=m&&k<=i;k++)if(!dir||k*2!=m){
			if(suf[0].query(n-i+1,n-i+k)==S.query(1,k)&&pre[1].query(i-k+1,i)==S.query(k+1,2*k))Add(f[1][i][2*k],1);
			if(suf[1].query(n-i+1,n-i+k)==S.query(1,k)&&pre[0].query(i-k+1,i)==S.query(k+1,2*k))Add(f[0][i][2*k],1);
		}
		f[0][i][0]=1,f[1][i][0]=1;
		for(int k=2;2*k<=m&&i+k<=n;k++)if(!dir||k*2!=m){
			if(pre[0].query(i+1,i+k)==S.query(m-2*k+1,m-k)&&suf[1].query(n-i-k+1,n-i)==S.query(m-k+1,m))Add(res,f[0][i][m-2*k]);
			if(pre[1].query(i+1,i+k)==S.query(m-2*k+1,m-k)&&suf[0].query(n-i-k+1,n-i)==S.query(m-k+1,m))Add(res,f[1][i][m-2*k]);
		}
		for(int j=0;j<m;j++){
			if(a[0][i+1]==s[j+1])Add(f[0][i+1][j+1],f[0][i][j]);
			if(a[1][i+1]==s[j+1])Add(f[1][i+1][j+1],f[1][i][j]);
			if(j+2<=m){
				if(a[1][i+1]==s[j+1]&&a[0][i+1]==s[j+2])Add(f[0][i+1][j+2],f[1][i][j]);
				if(a[0][i+1]==s[j+1]&&a[1][i+1]==s[j+2])Add(f[1][i+1][j+2],f[0][i][j]);
			}
		}
		Add(res,f[0][i][m]),Add(res,f[1][i][m]);
	}
	return res;
}
int ans;
int main(){
	init();
	scanf("%s",a[0]+1);
	scanf("%s",a[1]+1);
	n=strlen(a[0]+1);
	pre[0].init(a[0]);
	reverse(a[0]+1,a[0]+n+1);
	suf[0].init(a[0]);
	reverse(a[0]+1,a[0]+n+1);
	pre[1].init(a[1]);
	reverse(a[1]+1,a[1]+n+1);
	suf[1].init(a[1]);
	reverse(a[1]+1,a[1]+n+1);
	scanf("%s",s+1);
	m=strlen(s+1);
	S.init(s);
	Add(ans,calc(s,0));
	if(m==1){cout<<ans;return 0;}
	reverse(s+1,s+m+1);
	S.init(s);
	Add(ans,calc(s,1));
	if(m==2){
		for(int i=1;i<=n;i++){
			if(a[0][i]==s[1]&&a[1][i]==s[2])Dec(ans,1);
			if(a[1][i]==s[1]&&a[0][i]==s[2])Dec(ans,1);
		}
	}
	cout<<ans<<'
';
}
原文地址:https://www.cnblogs.com/stargazer-cyk/p/12328411.html