Codeforces346B Lucky Common Subsequence(KMP+dp)

题目链接

题解：对于LCS问题，我们的正解是用dp解决。而本题目中还有了一个数组是不能够包含的，所以我们可以给dp新增加一维。

dp[i][j][k]表示A串前长度为i的子串，B串前长度为j的子串，能匹配到的C串的最长位置为k时的最长长度。

转移时，与LCS问题类似。对于k的转移，就是要判断当前为是否可以继续匹配到C中k的下一位中去，如果可以，就加1，不可以，就根据失配函数进行转移，知道可以匹配或为起始位置为止（类似KMP的查找过程）。

而输出路径的话，就开另一个数组，记录下过程中每个点的来源，最后反向搜索即可。

PS：1.由于最后一位的存在，原来的的dp转移有了些变化。dp[i-1][j-1] + 1不一定大于dp[i-1][j],dp[i][j-1]，所以转移时要全部计算一边。

2.还有，由于k的存在，每一位都有可能单独成为一个起点，而此状态没有被转移。所以每次计算时，还要考虑从这个点开始的情况。

代码如下：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 100 + 10;
char A[maxn],B[maxn],C[maxn];
int f[maxn],dp[maxn][maxn][maxn];
int n,lena,lenb,lenc;
int path[maxn][maxn][maxn][4];
void getFail(char * P, int *f){
	int m = strlen(P);
	f[0] = 0;f[1] = 0;
	for(int i = 1;i < m;i++){
		 int j = f[i];
		 while(j && P[i] != P[j]) j = f[j];
		 f[i + 1] = P[i] == P[j] ? j + 1 : 0;
	}
}
int main()
{
	scanf("%s",A);
	scanf("%s",B);
	scanf("%s",C);
	getFail(C,f);
	lena = strlen(A);
	lenb = strlen(B);
	lenc = strlen(C);
	memset(dp,-1,sizeof(dp));
	memset(path,-1,sizeof(path));
	for(int i = 0;i < lena;i++) {
		if(A[i] == B[0]) {
			if(A[i] == C[0]) dp[i][0][1] = 1;
			else dp[i][0][0] = 1;
		}
		else dp[i][0][0] = 0;
		if(i > 0 && dp[i][0][1] == 0 && dp[i-1][0][1] > 0){
			dp[i][0][1] = dp[i-1][0][1];
			path[i][0][1][0] = i - 1;
			path[i][0][1][1] = 0;
			path[i][0][1][2] = 1;
		}
		if(i > 0 && dp[i][0][0] == 0 && dp[i-1][0][0] > 0){
			dp[i][0][0] = dp[i-1][0][0];
			path[i][0][0][0] = i - 1;
			path[i][0][0][1] = 0;
			path[i][0][0][2] = 0;
		}
	}
	for(int i = 0;i < lenb;i++) {
		if(A[0] == B[i]) {
			if(B[i] == C[0]) dp[0][i][1] = 1;
			else dp[0][i][0] = 1;
		}
		else dp[0][i][0] = 0;
		if(i > 0 && dp[0][i][1] == 0 && dp[0][i - 1][1] > 0){
			dp[0][i][1] = dp[0][i-1][1];
			path[0][i][1][0] = 0;
			path[0][i][1][1] = i-1;
			path[0][i][1][2] = 1;
		}
		if(i > 0 && dp[0][i][0] == 0 && dp[0][i-1][0] > 0){
			dp[0][i][0] = dp[0][i-1][0];
			path[0][i][0][0] = 0;
			path[0][i][0][1] = i-1;
			path[0][i][0][2] = 0;
		}	
	}
	
	for(int i = 1;i < lena;i++){
		for(int j = 1;j < lenb;j++){
			for(int k = 0;k < lenc;k++){
				if(A[i] == B[j] && dp[i - 1][j - 1][k] != -1) {
					//printf("lalal111\n");
					//if(dp[i - 1][j - 1][k] == -1) continue;
					int p = k;
					while(p && C[p] != A[i]) p = f[p];
					if(C[p] == A[i]) p++;
					if(dp[i - 1][j - 1][k] + 1 > dp[i][j][p]){
						dp[i][j][p] = dp[i - 1][j - 1][k] + 1;
						path[i][j][p][0] = i - 1;
						path[i][j][p][1] = j - 1;
						path[i][j][p][2] = k;
					}
				//	printf("i:  %d, j:  %d, p:  %d, dp:  %d\n",i,j,p,dp[i][j][p]);
				}
				if(A[i] == B[j] && k == 0 && A[i] != C[0]) {
					if(dp[i][j][k] < 1){
						dp[i][j][k] = 1;
						path[i][j][k][0] = -1;
						path[i][j][k][1] = -1;
						path[i][j][k][2] = -1;
					}
				}
				if(A[i] == B[j] && k == 1 && A[i] == C[0]) {
					if(dp[i][j][k] < 1){
						dp[i][j][k] = 1;
						path[i][j][k][0] = -1;
						path[i][j][k][1] = -1;
						path[i][j][k][2] = -1;
					}
				}
							
				//printf("lalal222\n");
				//if(dp[i-1][j][k] == -1 && dp[i][j - 1][k] == -1) continue;
				if(dp[i-1][j][k] != -1) {
					if(dp[i - 1][j][k] > dp[i][j][k]){
						dp[i][j][k] = dp[i - 1][j][k];
						path[i][j][k][0] = i - 1;
						path[i][j][k][1] = j;
						path[i][j][k][2] = k;
					}
				}
				if(dp[i][j - 1][k] != -1) {
					if(dp[i][j - 1][k] > dp[i][j][k]){
						dp[i][j][k] = dp[i][j - 1][k];
						path[i][j][k][0] = i;
						path[i][j][k][1] = j - 1;
						path[i][j][k][2] = k;
					}
				}
			//	printf("i:  %d, j:  %d, k:  %d, dp:  %d\n",i,j,k,dp[i][j][k]);
			}
		}
	}
	
	int ans = 0,p = 0;;
	for(int k = 0;k < lenc;k++){
		if(dp[lena - 1][lenb - 1][k] > ans){
			ans = dp[lena - 1][lenb - 1][k];
			p = k;
		}
	}
	int ii = lena - 1,jj = lenb - 1,kk = p;
	string anss;
	while(ii >= 0 && jj >= 0){
//		printf("ii:  %d, jj:  %d, kk:  %d\n",ii,jj,kk);cout << anss << endl;
		int x = path[ii][jj][kk][0];
		int y = path[ii][jj][kk][1];
		int z = path[ii][jj][kk][2];
//		printf("x:  %d, y:  %d, z:  %d\n",x,y,z);
		if(x < 0 || y < 0) {if(dp[ii][jj][kk] > 0) anss+=A[ii];break;}
		if(dp[ii][jj][kk] - dp[x][y][z] > 0) anss+=A[ii];
		ii = x;jj = y;kk = z;
	}
	reverse(anss.begin(),anss.end());
	if(ans != 0) cout << anss << endl;
	else printf("%d\n",ans);
	return 0;
}