小白紫书上动态规划专题的原题,书上有状态转移的代码,要自己写预处理
#include <iostream>
#include <stdio.h>
#include <string>
#include <cstring>
#include <algorithm>
#include <cmath>
#define INF 0x3f3f3f
using namespace std;
int vis[250][60][3];
int dp[250][60];
int t[220];
int d1[55],d2[55];
int n,T,m1,m2;
int main()
{
int ca=1;
while(~scanf("%d",&n))
{
if(n==0) break;
scanf("%d",&T);
for(int i=1;i<n;i++)
scanf("%d",&t[i]);
scanf("%d",&m1);
for(int i=1;i<=m1;i++)
scanf("%d",&d1[i]);
scanf("%d",&m2);
for(int i=1;i<=m2;i++)
scanf("%d",&d2[i]);
memset(vis,0,sizeof vis);
for(int i=1;i<=m1;i++)//从第一站向右开
{
vis[d1[i]][1][0]=1;
int temp=d1[i];
for(int j=1;j<n;j++)//i时刻j车站是否有向右的车
{
temp+=t[j];
if(temp<=T)
vis[temp][j+1][0]=1;
else
break;
}
}
for(int i=1;i<=m2;i++)//从最后一站向左
{
vis[d2[i]][n][1]=1;
int temp=d2[i];
for(int j=n-1;j>=1;j--)//i时刻j车站是否有车向左
{
temp+=t[j];
if(temp<=T)
vis[temp][j][1]=1;
else
break;
}
}
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&&vis[i][j][0]&&i+t[j]<=T)//如果选择向右的车(前提是有车)
dp[i][j]=min(dp[i][j],dp[i+t[j]][j+1]);
if(j>1&&vis[i][j][1]&&i+t[j-1]<=T)//如果选择向左的车(有车)
dp[i][j]=min(dp[i][j],dp[i+t[j-1]][j-1]);
}
}
printf("Case Number %d: ",ca++);
if(dp[0][1]>=INF) printf("impossible\n");
else
printf("%d\n",dp[0][1]);
}
return 0;
}