【noip 2015】字串 dp

一开始自己写了一个，不过感觉后面看别人写的要优美一些，都写把

定义dp[k][i][j]表示用A串的前i个元素的其中k段来组成B串的前j个的方案数

首先考虑On4的转移：

1.A[i]!=B[j] 毫无疑问 f[k][i][j]+=f[k][i-1][j]

2.A[i]==B[j]    那么现在就用很多选择了，例如：第i个元素单独作为一段，或者说A[i-1]==B[j-1]的话还可以和前一个元素一起组成新的一段，然后以此类推就好了所以转移方程:f[k][i][j]+=f[k-1][a][b]( a<i&&b<j&&A[a]==B[b])

好了这样我们有了第一个程序（未交到OJ测试）：

#include<cstdio>
#include<cstring>
#include<iostream>
#define LL long long
#define Mod 1000000007
using namespace std;
LL n,m,K,f[200][550][550],sum[220][550][550];
char a[2020],b[2020];

int main(){
	scanf("%lld%lld%lld",&n,&m,&K);
	scanf("%s",a+1),scanf("%s",b+1);
	for(LL i=0;i<=n;i++)f[0][i][0]=1;
	
	for(LL k=1;k<=K;k++){
		for(LL i=1;i<=n;i++){
			sum[k-1][i-1][0]=f[k-1][i-1][0];
			for(LL j=1;j<=m;j++){
				f[k][i][j]+=f[k][i-1][j];
				for(LL z=1;z<=j;z++){
					LL A=i-z+1,B=j-z+1;
					if(a[A]==b[B])f[k][i][j]+=f[k-1][A-1][B-1];// B:0->j-1
					else break;
				}
			}
		}
	}
	printf("%lld",f[K][n][m]);
	return 0;
}

但其实仔细想一想可以很简单有效且套路的优化，就是不必每一次都回去找，因为反正转移都是从1遍历过来的，所以可以维护一个前缀和，复杂度降至On3。额，我一开始就是直接写的On3的写法，所以没有多想结果因此忘记了讨论(A[i]!=B[j]&&A[i+1]==B[j+1])时候的情况，结果debug了好久，最后忍不了了就又跑去写了一个n4的，就有了上面那份代码，汗。。。。

时间能够接受了空间却很难接受，没关系每一次的转移只用到了k-1滚动起来就好了

#include<cstdio>
#include<cstring>
#include<iostream>
#define LL long long
#define Mod 1000000007
using namespace std;
LL n,m,K,f[2][2020][220],sum[2][2020][220];
char a[2020],b[2020];


int main(){
	scanf("%lld%lld%lld",&n,&m,&K);
	scanf("%s",a+1),scanf("%s",b+1);
	for(LL i=0;i<=n;i++)f[0][i][0]=1;
	
	LL pos=0;
	for(LL k=1;k<=K;k++){
		pos^=1;memset(f[pos],00,sizeof(f[pos])),memset(sum[pos^1],0,sizeof(sum[pos^1]));
		for(LL i=1;i<=n;i++){
			sum[pos^1][i-1][0]=f[pos^1][i-1][0];
			for(LL j=1;j<=m;j++){
				f[pos][i][j]=(f[pos][i][j]+f[pos][i-1][j])%Mod;
				if(a[i]==b[j]){
					sum[pos^1][i][j]=(sum[pos^1][i-1][j-1]+f[pos^1][i][j])%Mod;
					f[pos][i][j]=(f[pos][i][j]+sum[pos^1][i-1][j-1])%Mod;
				}
				if(a[i+1]==b[j+1]&&a[i]!=b[j])sum[pos^1][i][j]=f[pos^1][i][j];
			}
		}
	}
	printf("%lld",f[pos][n][m]);
	return 0;
}

还有一种定义方式增加一维表示对于A串的第i位到底选不选，定义f[i][j][k][0]表示A串前i位与B串前j位且必须以j结尾可以也可不以i结尾分成k段的方案数，f[i][j][k][1]则表示A串前i位且必须以A[i]结尾B也必须以B[j]结尾的方案数，就不用辅助数组了，方便的多

#include<cstdio>
#include<cstring>
#include<iostream>
#define Mod 1000000007
using namespace std;
int f[2][220][220][2],m,n,K,pos,last;
char a[1020],b[220];

int main(){
	scanf("%d%d%d%s%s",&n,&m,&K,a+1,b+1);
	f[pos][0][0][0]=f[pos^1][0][0][0]=1;
	for(int i=1;i<=n;i++){
		pos^=1,last=pos^1;
		for(int j=1;j<=m;j++){
			for(int k=1;k<=K;k++){
				if(a[i]==b[j])f[pos][j][k][1]=(f[last][j-1][k-1][0]+f[last][j-1][k][1])%Mod;
				else f[pos][j][k][1]=0;
				f[pos][j][k][0]=(f[pos][j][k][1]+f[last][j][k][0])%Mod;
			}
		}
	}
	printf("%d",f[pos][m][K][0]);
	return 0;
}