百练1661:帮助 Jimmy（动态规划）

总时间限制:
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
1
2
3
4
5
样例输出

23
注意： 关于MAX的限制，每一条都要判定是否大于MAX，是则置状态值为inf

#include<iostream>
#include<cstring>
#include<algorithm>
#include<set>
#include<map>
#include <vector>
#include <math.h>
#include <string>
using namespace std;
const int inf = 0x3f3f3f3f;
struct node{
    int x1,x2,h;
    bool operator < (const node& a)const{
        return h<a.h;
    }
};
vector<node> v;
int dp[1005][2];
int MAX;
int first_black(int n,int x)
{
    for(int i=n-1;i>=0;i--){
        if(x<=v[i].x2&&x>=v[i].x1)
            return i;
    }
    return inf;
}
int drop(int n)
{
    int m;
    for(int i=1;i<=n;i++){

        if((m=first_black(i,v[i].x1))!=inf){
            if(v[i].h-v[m].h<=MAX)
                dp[i][0]=v[i].h-v[m].h+min(dp[m][0]+v[i].x1-v[m].x1,dp[m][1]+v[m].x2-v[i].x1);
            else
                dp[i][0]=inf;
        }else{
            if(v[i].h<=MAX)
                dp[i][0]=v[i].h;
            else
                dp[i][0]=inf;
        }

        if((m=first_black(i,v[i].x2))!=inf){
            if(v[i].h-v[m].h<=MAX)
                dp[i][1]=v[i].h-v[m].h+min(dp[m][0]+v[i].x2-v[m].x1,dp[m][1]+v[m].x2-v[i].x2);
            else
                dp[i][1]=inf;
        }else{
            if(v[i].h<=MAX)
                dp[i][1]=v[i].h;
            else
                dp[i][1]=inf;
        }
    }
    return min(dp[n][0],dp[n][1]);
}

int main()
{
    int t;
    cin >> t;
    while(t--){
        memset(dp,0,sizeof(dp));
        v.clear();
        int N,X,Y;
        cin >> N >> X >> Y >> MAX;
        v.push_back(node{X,X,Y});
        for(int i=0;i<N;i++){
            node temp_node;
            cin >> temp_node.x1 >> temp_node.x2 >> temp_node.h;
            v.push_back(temp_node);
        }
        sort(v.begin(),v.end());
        dp[0][0]=dp[0][1]=v[0].h;
        cout << drop(N) << endl;
    }
    return 0;
}