原题指路
最小体力消耗路径
题目描述
你准备参加一场远足活动。给你一个二维 rows x columns
的地图 heights ,其中 heights[row][col]
表示格子 (row, col)
的高度。一开始你在最左上角的格子 (0, 0)
,且你希望去最右下角的格子 (rows-1, columns-1)
(注意下标从 0 开始编号)。你每次可以往 上,下,左,右 四个方向之一移动,你想要找到耗费 体力
最小的一条路径。
一条路径耗费的 体力值
是路径上相邻格子之间 高度差绝对值
的 最大值
决定的。
请你返回从左上角走到右下角的最小 体力消耗值
。
解题思路
刚开始看这道题的时候有点发懵,于是轻车熟路地点开了官方题解(官方题解)。然后发现···原来把题目抽象成图,然后用Dijkstra算法就可以算。
思路:
1、将二维的坐标一维化:即将(x,y)记为x*col+y;
2、使用优先队列对最短路径进行排序上的控制;
3、Dijkstra算法构造:这道题可以想成BFS+优先队列,当然这里Dijkstra的更新条件记得要改一下。
代码
class Dist//定义图的节点
{
public:
int x,y,h;
Dist(int a,int b,int c):x(a),y(b),h(c){}
bool operator < (const Dist& n) const
{
return h>n.h;
}
};
class Solution
{
private:
int dir[4][2]={{0,1},{0,-1},{1,0},{-1,0}};//定义行走的方向
public:
int minimumEffortPath(vector<vector<int>>& heights)
{
int row=heights.size(),col=heights[0].size();
priority_queue<Dist> q;//用优先队列进行排序的控制
vector<bool> visited(row*col);//记录已经更新过的节点
map<int,int> dist;//记录到每个节点的最短路径
q.emplace(0,0,0);
dist[0]=0;
while(!q.empty())//Dijkstra算法
{
auto [x,y,h]=q.top();
q.pop();
int id=x*col+y;
if(visited[id])
continue;
visited[id]=true;
dist[id]=h;
for(int i=0;i<4;i++)
{
int nx=x+dir[i][0];
int ny=y+dir[i][1];
if(nx>=0&&nx<row&&ny>=0&&ny<col&&!visited[nx*col+ny])
q.emplace(nx,ny,max(h,abs(heights[x][y]-heights[nx][ny])));
}
}
return dist[row*col-1];
}
};