poj 1661 Help Jimmy 动态规划

题意：有n个平台，每个平台给出左端点x1,右端点x2,高度h，jimmy从坐标（X，Y）下落，下落速度和平台移动速度为1m/s，最大下落高度不超过mmax，求到地面的最短时间。

按平台高度排序，从高到低遍历。若平台i上有平台j的高度差小于mmax，左端点或右端点没有被i与j之间的平台覆盖，且范围在平台I的区间中，则更新掉到平台i左右端点的时间，并覆盖平台j的左端点或右端点。d[i][0],d[i][1]分别代表从起始点到平台i左右端点的最短时间。v[i][0],v[i][1]分别代表平台i的左右端点是否被覆盖。

#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 1100
#define INF 0x7ffffff

using namespace std;

int d[N][2],v[N][2];

struct node
{
    int x1,x2,h;
}p[N];

bool cmp(node a,node b)
{
    return a.h>b.h;
}

int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        int n,X,Y,mmax,ans=INF;
        cin>>n>>X>>Y>>mmax;
        p[0].x1=p[0].x2=X;
        p[0].h=Y;
        d[0][0]=d[0][1]=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d%d",&p[i].x1,&p[i].x2,&p[i].h);
            if(p[i].x1>p[i].x2) swap(p[i].x1,p[i].x2);
            d[i][0]=d[i][1]=INF;
        }
        sort(p,p+n+1,cmp);
        memset(v,0,sizeof(v));
        for(int i=0;i<=n;i++)
        {
            for(int j=0;j<i;j++)
            {
                if(p[i].h<p[j].h&&p[j].h-p[i].h<=mmax)
                {
                    if(!v[j][0]&&p[i].x1<=p[j].x1&&p[i].x2>=p[j].x1)
                    {
                        d[i][0]=min(d[i][0],d[j][0]+p[j].h-p[i].h+p[j].x1-p[i].x1);
                        d[i][1]=min(d[i][1],d[j][0]+p[j].h-p[i].h+p[i].x2-p[j].x1);
                        v[j][0]=1;
                    }
                    if(!v[j][1]&&p[i].x1<=p[j].x2&&p[i].x2>=p[j].x2)
                    {
                        d[i][0]=min(d[i][0],d[j][1]+p[j].h-p[i].h+p[j].x2-p[i].x1);
                        d[i][1]=min(d[i][1],d[j][1]+p[j].h-p[i].h+p[i].x2-p[j].x2);
                        v[j][1]=1;
                    }
                }
            }
        }
        for(int i=0;i<=n;i++)
        {
            if(p[i].h<=mmax)
            {
                if(!v[i][0])    ans=min(ans,d[i][0]+p[i].h);
                if(!v[i][1])    ans=min(ans,d[i][1]+p[i].h);
            }
        }
        cout<<ans<<endl;
    }
}