poj 1661 help Jimmy(dp)

poj 1661 help Jimmy(dp) 程序设计实习动态规划练习
总时间限制: 1000ms 内存限制: 65536kB

描述
“Help Jimmy” 是在下图所示的场景上完成的游戏。

场景中包括多个长度和高度各不相同的平台。地面是最低的平台，高度为零，长度无限。

Jimmy老鼠在时刻0从高于所有平台的某处开始下落，它的下落速度始终为1米/秒。当Jimmy落到某个平台上时，游戏者选择让它向左还是向右跑，它跑动的速度也是1米/秒。当Jimmy跑到平台的边缘时，开始继续下落。Jimmy每次下落的高度不能超过MAX米，不然就会摔死，游戏也会结束。

设计一个程序，计算Jimmy到底地面时可能的最早时间。

输入
第一行是测试数据的组数t（0 <= t <= 20）。每组测试数据的第一行是四个整数N，X，Y，MAX，用空格分隔。N是平台的数目（不包括地面），X和Y是Jimmy开始下落的位置的横竖坐标，MAX是一次下落的最大高度。接下来的N行每行描述一个平台，包括三个整数，X1[i]，X2[i]和H[i]。H[i]表示平台的高度，X1[i]和X2[i]表示平台左右端点的横坐标。1 <= N <= 1000，-20000 <= X, X1[i], X2[i] <= 20000，0 < H[i] < Y <= 20000（i = 1..N）。所有坐标的单位都是米。

Jimmy的大小和平台的厚度均忽略不计。如果Jimmy恰好落在某个平台的边缘，被视为落在平台上。所有的平台均不重叠或相连。测试数据保证问题一定有解。

输出
对输入的每组测试数据，输出一个整数，Jimmy到底地面时可能的最早时间。

样例输入

1
3 8 17 20
0 10 8
0 10 13
4 14 3

样例输出

23

来源
POJ Monthly–2004.05.15 CEOI 2000

---

本题原理明白了，就是动态规划，但是一直WA。
poj的discuss上讲的WA原因似乎有直接落地没考虑，没考虑平齐问题，对着这些一一调试了一些错误，但是还是WA。
copy了别人的代码，发现别人的代码比自己简洁很多（这就减少了WA的可能性），主要原因如下：别人把出发点看作一个特殊的平台，简化了模型，使不同情况合并。而且别人用非递归过程实现dp，既减少函数递归调用开销又使逻辑清晰。

---

期末复习时候又重做此题目，把参考代码自己了一下，有必要好好体会参考代码的技巧令代码简洁（6.20）

---

参考代码

#include <stdio.h>
#include <stdlib.h>

#define MAXN 1010
#define INF 9000000

typedef struct platform{
    int x1;
    int x2;
    int high;
}Platform;

int compare (const void * p, const void * q){
    Platform * p1 = (Platform *)p;
    Platform * q1 = (Platform *)q;
    return p1->high - q1->high;
}

int Min (int a, int b){
    return (a < b) ? a : b;
}

int N, X, Y, MAX;
Platform plat[MAXN];
int dp[MAXN][2];         //dp[i][0]、dp[i][1]分别表示从第i个平台左、右边到地面的最短时间

void LeftMinTime (int i){    //计算从平台i左边到地面的最短时间
    int k = i - 1;
    while (k > 0 && plat[i].high - plat[k].high <= MAX){
        //如果平台i左边下面有平台，且两者相距不超过MAX
        if (plat[i].x1 >= plat[k].x1 && plat[i].x1 <= plat[k].x2){
            dp[i][0] = plat[i].high - plat[k].high + 
                Min (plat[i].x1 - plat[k].x1 + dp[k][0], plat[k].x2 - plat[i].x1 + dp[k][1]);
            return;
        }
        else
            --k;
    }
    //如果平台i左边下面没有平台，或者两者相距超过了MAX
    if (plat[i].high - plat[k].high > MAX)
        dp[i][0] = INF;
    else
        dp[i][0] = plat[i].high;
}

void RightMinTime (int i){   //计算从平台i右边到地面的最短时间
    int k = i - 1;
    while (k > 0 && plat[i].high - plat[k].high <= MAX){
        //如果平台i右边下面有平台，且两者相距不超过MAX
        if (plat[i].x2 >= plat[k].x1 && plat[i].x2 <= plat[k].x2){
            dp[i][1] = plat[i].high - plat[k].high + 
                Min (plat[i].x2 - plat[k].x1 + dp[k][0], plat[k].x2 - plat[i].x2 + dp[k][1]);
            return;
        }
        else
            --k;
    }
    //如果平台i右边下面没有平台，或者两者相距超过了MAX
    if (plat[i].high - plat[k].high > MAX)
        dp[i][1] = INF;
    else
        dp[i][1] = plat[i].high;
}

int ShortestTime (){
    int i, j;

    for (i=1; i<=N+1; ++i){
        LeftMinTime (i);
        RightMinTime (i);
    }
    return Min (dp[N+1][0], dp[N+1][1]);
}

int main(){
    int t;
    int i;
    while (scanf ("%d", &t) != EOF){
        while (t-- != 0){
            scanf ("%d%d%d%d", &N, &X, &Y, &MAX);
            for (i=1; i<=N; ++i){
                scanf ("%d%d%d", &plat[i].x1, &plat[i].x2, &plat[i].high);
            }
            plat[0].high = 0;
            plat[0].x1 = -20000;
            plat[0].x2 = 20000;
            plat[N+1].high = Y;
            plat[N+1].x1 = X;
            plat[N+1].x2 = X;
            //根据平台高度按从低到高排序
            qsort (plat, N+2, sizeof(Platform), compare);
            printf ("%d\n", ShortestTime());
        }
    }

    return 0;
}

我的代码

#define TEST
#undef TEST

#define MAX_N 1000
#define INF 40000000

#include<stdio.h>
#include<stdlib.h>

struct platform_type
{
    int x1,x2,h;
};

platform_type p[MAX_N+2];
int cases,n,max_h;
int left[MAX_N+2],right[MAX_N+2];

int Min(int a,int b)
{
    return a<b?a:b;
}

int compare(const void* e1,const void* e2)
{
    return ((const platform_type*)e1)->h-((const platform_type*)e2)->h;
}

int main()
{
    int k;
    scanf("%d",&cases);
    p[0].x1=-20000;
    p[0].x2=20000;
    p[0].h=0;
    while (cases--)
    {
        scanf("%d",&n);
        scanf("%d%d%d",&p[n+1].x1,&p[n+1].h,&max_h);
        p[n+1].x2=p[n+1].x1;    
        for (int i=1;i<=n;i++)
            scanf("%d%d%d",&p[i].x1,&p[i].x2,&p[i].h);
        qsort(p,n+2,sizeof(platform_type),compare);
        #ifdef TEST
        for (int i=0;i<=n+1;i++)
            printf("%d %d %d\n",p[i].x1,p[i].x2,p[i].h);
        #endif 
        for (int i=1;i<=n+1;i++)
        {
            k=i-1;
            left[i]=-1;
            while (k && p[i].h-p[k].h<=max_h)
            {
                if (p[i].x1>=p[k].x1 && p[i].x1<=p[k].x2)
                {
                    left[i]=p[i].h-p[k].h+
                            Min(left[k]+p[i].x1-p[k].x1,right[k]+p[k].x2-p[i].x1);
                    break;
                }
                else
                    k--;
            }
            if (left[i]==-1)
            {
                if (p[i].h-p[k].h>max_h)
                    left[i]=INF;
                else
                    left[i]=p[i].h;
            }
            k=i-1;
            right[i]=-1;
            while (k && p[i].h-p[k].h<=max_h)
            {
                if (p[i].x2>=p[k].x1 && p[i].x2<=p[k].x2)
                {
                    right[i]=p[i].h-p[k].h+
                            Min(left[k]+p[i].x2-p[k].x1,right[k]+p[k].x2-p[i].x2);
                    break;
                }
                else
                    k--;
            }
            if (right[i]==-1)
            {
                if (p[i].h-p[k].h>max_h)
                    right[i]=INF;
                else
                    right[i]=p[i].h;
            }
        }
        printf("%d\n",left[n+1]);
    }
    return 0;
}