POJ 1661 Help Jimmy

点击打开链接

#include <iostream>
#include <algorithm>
using namespace std;
const int M =21000;
const int Inf=1<<30;

typedef struct{
int X;
int Y;
int H;
}Node;
Node node[M];
int n;
bool operator<(const Node a,const Node b)
{
return a.H>b.H;
}
int main()
{
int t;
cin>>t;
while(t--)
{
int x,y,MAX;
cin>>n>>x>>y>>MAX;
int R[M],L[M];// R[i] 从i号平台右端下落到地面的最短时间
node[0].X=node[0].Y=x;
node[0].H=y; // 初始位置

for(int i=1;i<=n;i++)
{
cin>>node[i].X>>node[i].Y>>node[i].H;
}
sort(node,node+n+1);

for(int i=n;i>=0;i--) // 逆推
{
int j;
for(j=i+1;j<=n;j++) // 找 i的左端点下发有plant吗
{
if(node[i].X>=node[j].X&&node[i].X<=node[j].Y)
{
break;
}
}
if(j>n)
{
if(node[i].H>MAX)
{
L[i]=Inf;
}
else
L[i]=node[i].H;
}
else
{
int k=node[i].H-node[j].H;
if(k>MAX) L[i]=Inf;
else
L[i]=min(node[i].X-node[j].X+L[j],node[j].Y-node[i].X+R[j])+k; // dp到J后选择方向
}
for(j=i+1;j<=n;j++) // 找 i的右端点下发有plant吗
{
if(node[i].Y>=node[j].X&&node[i].Y<=node[j].Y)
{
break;
}
}
if(j>n)
{
if(node[i].H>MAX)
{
R[i]=Inf;
}
else
R[i]=node[i].H;
}
else
{
int k=node[i].H-node[j].H;
if(k>MAX) //
R[i]=Inf;
else
R[i]=min(node[i].Y-node[j].X+L[j],node[j].Y-node[i].Y+R[j])+k; // dp:到J后选择方向
}

}
int ans=min(L[0],R[0]);
cout<<ans<<endl;
}
return 0;
}