【Codeforces 613E】—Puzzle Lover(DP)

mac2022-06-30  28

传送门

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

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

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

具体看代码(太麻烦了)

#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<<'\n'; }
最新回复(0)