UVA 1025 A Spy in the Metro - 简单dp

题目描述

分析：

关键信息：车站，时刻
dp[j][i] ：在j时刻，正处在i车站，需要等待的最短时间。
dp[j][i]=min{
(1). dp[j-1][i]+1 |等待1分钟 ,
(2). dp[j-cost(i+1,i)][i+1] |在j-cost(i+1,i)有i+1到i的车,
(3). dp[j-cost(i-1,i)][i-1] |在j-cost(i-1,i)有i-1到i的车
}

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define MAXN 50
#define MAXT 200
#define MAXM 50
#define INF 2000000000

int n,T,m1,m2,t[MAXN+10],l[MAXM+10],r[MAXM+10],dp[MAXT+10][MAXN+10];
bool train[MAXT+10][MAXN+10][MAXN+10];
void read()
{
    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",&l[i]);
    scanf("%d",&m2);
    for(int i=1;i<=m2;i++)
        scanf("%d",&r[i]);
}
void prepare()
{
    memset(train,0,sizeof train);
    for(int i=1;i<=m1;i++)
        for(int j=1,p=l[i];j<n;j++){
            p+=t[j];
            if(p>T)
                continue;
            train[p][j+1][j]=true;
        }
    for(int i=1;i<=m2;i++)
        for(int j=n,p=r[i];j>1;j--){
            p+=t[j-1];
            if(p>T)
                continue;
            train[p][j-1][j]=true;
        }
}
void DP()
{
    for(int i=0;i<=T;i++)
        for(int j=1;j<=n;j++)
            dp[i][j]=INF;
    dp[T][n]=0;
    for(int i=T-1;i>=0;i--)
        for(int j=1;j<=n;j++){
            if(dp[i+1][j]!=INF)
                dp[i][j]=dp[i+1][j]+1;
            if(i+t[j]<=T&&j+1<=n&&train[i+t[j]][j+1][j])
                dp[i][j]=min(dp[i][j],dp[i+t[j]][j+1]);
            if(i+t[j-1]<=T&&j-1>=1&&train[i+t[j-1]][j-1][j])
                dp[i][j]=min(dp[i][j],dp[i+t[j-1]][j-1]);
        }
    if(dp[0][1]==INF)
        printf("impossible\n");
    else
        printf("%d\n",dp[0][1]);
}
int main()
{
    //freopen("my.out","w",stdout);
    int cas=0;
    while(scanf("%d",&n)&&n){
        read();
        prepare();
        printf("Case Number %d: ",++cas);
        DP();
    }
}