UVALive 3605 & ZOJ 3135 Roommate（DP）

题目链接：点击打开链接

2700+ms过的。。感觉应该还有优化的空间。

设dp[i][j][k]为，当前第一个人已经使用了前i个设备，第二个人使用了前j个设备，且第一个人最后一个设备用完的时刻减去第二个人最后一个设备用完的时刻的时间差为k。

状态的转移的话，就只有加入第一个人的设备（dp[i+1][j]）和加入第二个人的设备（dp[i][j+1]）两种选择。

首先，从贪心的角度想，k这个只能是单一设备使用时产生的。所以当k为正的时候只能再加入第一个人的设备，当k为负的时候只能再加入第二个人的设备。

那么k的范围就被限定在-100到100之间了，再加个常数就能够成为数组的下标了。

其次，dp[i][j][k]的取值可以是绝对值比k更大的dp[i][j][k]，这个详见代码的第49行~第52行。

最后，由于开不出300*300*200的空间，所以需要将某一维改为滚动数组，这里改的是第一维。

代码如下：

#include<bits/stdc++.h>
#define FIX 105
using namespace std;

int dp[2][305][305];

int Time[2][55];

int v[2][305];

int getlen(int i,int j)
{
//	printf("Getlen(%d,%d)=%d\n",i,j,Time[i][v[i][j]]);
	return Time[i][v[i][j]];
}

void update(int &a,int p)
{
	if(p==-1)return;
	if(a==-1)a=p;
	else a=min(a,p);
}

int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		int n;
		scanf("%d",&n);
		for(int i=0;i<2;i++)
			for(int j=1;j<=n;j++)
				scanf("%d",&Time[i][j]);
		int a,b;
		scanf("%d%d",&a,&b);
		for(int i=1;i<=a;i++)
			scanf("%d",&v[0][i]);
		for(int j=1;j<=b;j++)
			scanf("%d",&v[1][j]);
		memset(dp,-1,sizeof(dp));
		dp[0][0][0+FIX]=0;
		int cur=0;
		for(int i=0;i<=a;i++)
		{
			memset(dp[cur^1],-1,sizeof(dp[cur^1]));
			for(int j=0;j<=b;j++)
			{
				for(int k=-100+FIX;k<FIX;k++)
					update(dp[cur][j][k+1],dp[cur][j][k]);
				for(int k=100+FIX;k>FIX;k--)
					update(dp[cur][j][k-1],dp[cur][j][k]);
				for(int k=-100+FIX;k<=100+FIX;k++)
				{
					if(dp[cur][j][k]==-1)continue;
			//		printf("dp[%d][%d][%d]=%d(%d)\n",i,j,k-FIX,dp[cur][j][k],cur);
					if(i!=a&&k<=FIX)
					{
						if(v[0][i+1]==v[1][j])
							update(dp[cur^1][j][getlen(0,i+1)+FIX],dp[cur][j][k]+getlen(0,i+1));
						else
							update(dp[cur^1][j][getlen(0,i+1)+k],dp[cur][j][k]+max(0,getlen(0,i+1)+k-FIX));
					}
					if(j!=b&&k>=FIX)
					{
						if(v[1][j+1]==v[0][i])
							update(dp[cur][j+1][-1*getlen(1,j+1)+FIX],dp[cur][j][k]+getlen(1,j+1));
						else
							update(dp[cur][j+1][k-getlen(1,j+1)],dp[cur][j][k]+max(0,getlen(1,j+1)-(k-FIX)));
					}
				}
			}
			cur^=1;
		}
		cur^=1;
		int ans=-1;
		for(int i=-100+FIX;i<=100+FIX;i++)
			update(ans,dp[cur][b][i]);
		printf("%d\n",ans);
	}
	return 0;
}