一道dp的题 :
dp[i][j] 表示从在时刻i 处在j站台 所需要最少还要等待多少时间
此时有3中决策
1.等1分钟
2.上开往右侧的火车(如果有的话)
3.上开往左侧的火车(如果有的话)
2.上开往右侧的火车(如果有的话)
3.上开往左侧的火车(如果有的话)
边界条件 dp[T][n] = 0; for( i = 1 to n - 1) dp[T][i] = INF;
要求出 dp[0][1]
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
int n,T;
int t[100];
int dp[205][55];
int M1,M2;
int d[55];
int e[55];
bool has_train[205][55][2];
void process()
{
for(int j = 1; j < n; ++j)
{
int sum = 0;
for(int k = 1;k <= j;++k)
{
sum += t[k - 1];
}
for(int i = 0; i < M1; ++i)
{
if(d[i] + sum <= T)
has_train[d[i] + sum][j][0] = true;
}
}
for(int j = n;j > 1;--j)
{
int sum = 0;
for(int k = n;k >= j;--k)
sum += t[k];
for(int i = 0;i < M2;++i)
{
if(e[i] + sum <= T)
has_train[e[i] + sum][j][1] = true;
}
}
}
int main()
{
int kase = 0;
while(~scanf("%d",&n) && n)
{
scanf("%d",&T);
memset(dp,0,sizeof(dp));
memset(has_train,false,sizeof(has_train));
memset(t,0,sizeof(t));
for(int i = 1; i < n; ++i)
scanf("%d",&t[i]);
scanf("%d",&M1);
for(int i = 0; i < M1; ++i)
scanf("%d",&d[i]);
scanf("%d",&M2);
for(int i = 0; i < M2; ++i)
scanf("%d",&e[i]);
process();
for(int i = 1;i < n;++i)
dp[T][i] = INF;
dp[T][n] = 0;
for(int i = T - 1;i >= 0;--i)
{
for(int j = 1;j <= n;++j)
{
dp[i][j] = dp[i + 1][j] + 1; // 原地等待
if(j < n && has_train[i][j][0] && i + t[j] <= T) //坐向右的车
dp[i][j] = min(dp[i][j],dp[i + t[j]][j + 1]);
if(j > 1 && has_train[i][j][1] && i + t[j - 1] <= T) //坐向左的车
dp[i][j] = min(dp[i][j],dp[i + t[j - 1]][j - 1]);
}
}
if(dp[0][1] >= INF)
printf("Case Number %d: impossible\n",++kase);
else
printf("Case Number %d: %d\n",++kase,dp[0][1]);
}
return 0;
}