BZOJ 5107 找爸爸 DP

BZOJ 5107
题意:给出两个长度分别为n,m的字符串a,b.字符串只包含{A,T,C,G}.
操作:在两个序列的任意位置插入任意多的空格,使得两个字符串长度相同.
操作结束后进行匹配,计算价值:若当前比较位都不是空格 则价值+=d[a[i]][b[i]] 
并定义g(k)=-A-B(k-1) k为a,b中连续空格的个数 最后价值在加上所有的g(k).


0<B<A<=1000, -1000<=d(x,y)<=1000, n+m<=3000.问操作过后 最大能获得的价值为?
样例
ATGG
ATCC
5 -4 -4 -4
-4 5 -4 -4
-4 -4 5 -4
-4 -4 -4 5
2 1


操作后:{ATGG--},{AT--CC}.
ans=4;


n+m<=3000 可以设状态a的前i个字符和b的前j个字符在操作后能得到的最大值
因为d[i][j]匹配时.最后一列只有三种情况:{a[i],b[j]} {a[i],space},{space,b[j]}

不会出现两个空格匹配 因为可以删除这两个空格使得答案增大.

设dp[i][j][0/1][0/1] 还要知道连续空格数,记录匹配最后一列的状态 暴力进行转移即可.

#include <bits/stdc++.h>
using namespace std;
const int N=3e3+5,inf=1e9;
char a[N],b[N];
int d[5][5],A,B,mp[N];
int f[N][N][2][2];
int main()
{
	mp['A']=1,mp['T']=2,mp['G']=3,mp['C']=4;
	scanf("%s%s",a+1,b+1);
	for(int i=1;i<=4;i++)
		for(int j=1;j<=4;j++)
			scanf("%d",&d[i][j]);
	scanf("%d%d",&A,&B);
	int n=strlen(a+1),m=strlen(b+1);
	for(int i=0;i<=n;i++)
		for(int j=0;j<=m;j++)
			f[i][j][0][0]=f[i][j][0][1]=f[i][j][1][0]=-inf;
	f[0][0][0][0]=0;
	//------1
	//.....j 
	for(int i=0;i<=n;i++)
	{
		for(int j=0;j<=m;j++)
		{
			if(i==j&&i==0)
				continue;
			int x,y;
			if(i)
				x=mp[a[i]];
			if(j)
				y=mp[b[j]];
			
			if(i&&j)
			{
				f[i][j][0][0]=max(f[i][j][0][0],f[i-1][j-1][0][0]+d[x][y]);
				f[i][j][0][0]=max(f[i][j][0][0],f[i-1][j-1][0][1]+d[x][y]);	
				f[i][j][0][0]=max(f[i][j][0][0],f[i-1][j-1][1][0]+d[x][y]);
			}
			if(i)
			{
			
				f[i][j][0][1]=max(f[i][j][0][1],f[i-1][j][0][0]-A);
				f[i][j][0][1]=max(f[i][j][0][1],f[i-1][j][0][1]-B);
				f[i][j][0][1]=max(f[i][j][0][1],f[i-1][j][1][0]-A);
			}
			if(j)
			{  
				f[i][j][1][0]=max(f[i][j][1][0],f[i][j-1][0][0]-A);
				f[i][j][1][0]=max(f[i][j][1][0],f[i][j-1][0][1]-A);
				f[i][j][1][0]=max(f[i][j][1][0],f[i][j-1][1][0]-B);
			}
		}
	}
	int ans=max(max(f[n][m][0][0],f[n][m][0][1]),f[n][m][1][0]);
	cout<<ans<<endl;
	return 0;	
}