Help Jimmy POJ - 1661 dp

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=;
const int M=;
const int INF=0x3f3f3f;
struct Time
{
    int x1,x2,h;
}a[N];
int dp[N+][]; // 0：表示第i的木板左边到底部的最短时间
              // 1: 表示第i的木板右边到底部的最短时间
int n,x,y,max_h;
int cmp(Time c,Time b) //排序函数
{
    return c.h > b.h; //从大到小排列
}
void left(int i)//左
{
    //如果平台i下面有平台，且两者相距不超过MAX
    int k = i+;
    while( k<n+ && a[i].h-a[k].h<=max_h ) //n+1处是地面
    {
        //确保平台k在平台i的左下方，且小鼠可以跳到上面
        if(a[i].x1>=a[k].x1&&a[i].x1<= a[k].x2)
        {
            //从i的左端跳到地面的时间
            //i先跳到k的时间+
            //min(从k左端跳到地面的时间+从i的左端到k的左端的时间，从k的右端跳到地面的时间+从i左端到k的右端)
            dp[i][]=a[i].h-a[k].h+min(dp[k][]+a[i].x1-a[k].x1,dp[k][]+a[k].x2-a[i].x1);
            return;
        }
        k++;
    }
    //因为第二个条件出的循环即：不能到达下一平台
    if(a[i].h-a[k].h>max_h)
        dp[i][]=INF;
    //因为它下面没木板，直接落地
    else
        dp[i][]=a[i].h;
}
void right(int i)//右
{
    //如果平台i下面有平台，且两者相距不超过MAX
    int k=i+;
    while(k<n+ && a[i].h-a[k].h<=max_h)
    {
        //确保平台k在平台i的右下方，且小鼠可以跳到上面
        if(a[i].x2<=a[k].x2 && a[i].x2 >= a[k].x1)
        {
            dp[i][]=a[i].h-a[k].h+min(dp[k][] + a[i].x2-a[k].x1  ,  dp[k][] + a[k].x2-a[i].x2);
            return;
        }
        k++;
    }
    //不能到达下一平台
    if(a[i].h-a[k].h > max_h)
        dp[i][]=INF;
    //因为它下面没木板，直接落地
    else
        dp[i][]=a[i].h;
}
int main()
{
    int t,i;
    cin>>t;
    while(t--)
    {
        //木板个数、老鼠的初始坐标、最大跳跃高度
        cin>>n>>x>>y>>max_h;
        //地面也当做一块木板
        a[].x1=-;
        a[].x2=;
        a[].h=;
        //老鼠初始位置也当做一块木板
        a[].x1=x;
        a[].x2=x;
        a[].h=y;
        //输入数据
        for(i=;i<=n+;i++)
            cin>>a[i].x1>>a[i].x2>>a[i].h;
        //高度从大到小
        sort(a,a+n+,cmp);
        //初始化
        memset(dp,,sizeof(dp));
        //n+1是地面
        //所以从n开始
        for(i=n; i>= ;i--)
        {
            //进行左：去bfs
            left(i);
            //进行右：去bfs
            right(i);
        }
        cout<<min(dp[][],dp[][])<<endl;
    }
    return ;
}