最小体力消耗路径

原题指路
最小体力消耗路径

题目描述

你准备参加一场远足活动。给你一个二维 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];
        }
};