UVA 1025 A Spy in the Metro

小白紫书上动态规划专题的原题,书上有状态转移的代码,要自己写预处理

#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;
}


你可能感兴趣的:(UVA 1025 A Spy in the Metro)