POJ--1661--Help Jimmy--DP

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define INF 99999999
#define maxn   1005
using namespace std;
struct board
{
int x1,x2;	//板子的左右端点
int h;		//板子的高度
int lminTime,rminTime;	//分别从板子的左边和右边落到地上的最短时间
}dp[maxn];
int ox,oy;	//the original position
int mh;		//the max height he can jump
int n;		//the number of the board

void init()
{
scanf("%d%d%d%d",&n,&ox,&oy,&mh);
for(int i=0;i<n;i++)
{
scanf("%d%d%d",&dp[i].x1,&dp[i].x2,&dp[i].h);
dp[i].lminTime=dp[i].rminTime=INF;
}
}
int cmp(board a,board b)
{
return a.h>b.h;
}
void solve()
{
bool lb,rb;
sort(dp,dp+n,cmp);
int temph;
for(int i=n-1;i>=0;i--)
{
lb=false;
rb=false;
for(int j=i+1;j<n;j++)
{
if(dp[i].h==dp[j].h)continue;
temph=dp[i].h-dp[j].h;
if(temph>mh)continue;
if(lb==false&&dp[i].x1>=dp[j].x1&&dp[i].x1<=dp[j].x2)
{
lb=true;	//注意一旦有一个板子能接住，那么下面的板子就不能再接这个地方落下来的小人了
dp[i].lminTime=min(dp[i].lminTime,temph+dp[j].lminTime+dp[i].x1-dp[j].x1);
dp[i].lminTime=min(dp[i].lminTime,temph+dp[j].rminTime+dp[j].x2-dp[i].x1);
}
if(rb==false&&dp[i].x2>=dp[j].x1&&dp[i].x2<=dp[j].x2)
{
rb=true;	//同上
dp[i].rminTime=min(dp[i].rminTime,temph+dp[i].x2-dp[j].x1+dp[j].lminTime);
dp[i].rminTime=min(dp[i].rminTime,temph+dp[j].x2-dp[i].x2+dp[j].rminTime);
}
}
if(dp[i].h<=mh)
{//如果左边或者右边没有板子能接住小人的话，就直接落到地上
if(lb==false)
{
dp[i].lminTime=min(dp[i].lminTime,dp[i].h);
}
if(rb==false)
{
dp[i].rminTime=min(dp[i].rminTime,dp[i].h);
}
}
}
int ans=INF;
bool jp=false;
for(int i=0;i<n;i++)
{
if(oy<dp[i].h)continue;
if(dp[i].x1<=ox&&dp[i].x2>=ox)
{
jp=true; //同理，找到第一个能接住小人的板子就不往下找了
ans=min(ans,oy-dp[i].h+ox-dp[i].x1+dp[i].lminTime);
ans=min(ans,oy-dp[i].h+dp[i].x2-ox+dp[i].rminTime);
break;
}
}
if(jp)
printf("%d\n",ans);
else
printf("%d\n",oy);
}
int main()
{
//freopen("1661.txt","r",stdin);
int t;
scanf("%d",&t);
while(t--)
{
init();
solve();
}
return 0;
}