letecode 编程学习（34）

题目

你准备参加一场远足活动。给你一个二维 rows x columns 的地图 heights ，其中 heights[row][col] 表示格子 (row, col) 的高度。一开始你在最左上角的格子 (0, 0) ，且你希望去最右下角的格子 (rows-1, columns-1) （注意下标从 0 开始编号）。你每次可以往 上，下，左，右 四个方向之一移动，你想要找到耗费 体力 最小的一条路径。

一条路径耗费的 体力值 是路径上相邻格子之间 高度差绝对值 的 最大值 决定的。

请你返回从左上角走到右下角的最小 体力消耗值 。

 

示例 1：

输入：heights = [[1,2,2],[3,8,2],[5,3,5]]
输出：2
解释：路径 [1,3,5,3,5] 连续格子的差值绝对值最大为 2 。
这条路径比路径 [1,2,2,2,5] 更优，因为另一条路径差值最大值为 3 。
示例 2：

输入：heights = [[1,2,3],[3,8,4],[5,3,5]]
输出：1
解释：路径 [1,2,3,4,5] 的相邻格子差值绝对值最大为 1 ，比路径 [1,3,5,3,5] 更优。
示例 3：

输入：heights = [[1,2,1,1,1],[1,2,1,2,1],[1,2,1,2,1],[1,2,1,2,1],[1,1,1,2,1]]
输出：0
解释：上图所示路径不需要消耗任何体力。

解题思路

假设数组大小为m行n列，这个时候对数组的每一个元素进行编号0到m*n-1,

将整个数组看成一个无向图，并将边按照从小到大排列，分别将边上点对加入并查集中

如果0号和m*n-1号点有公共的祖先节点的时候，说明这个路径就是最小的路径。

代码

class Solution {
public:
    vector<int> pa;
    vector<int> rank;
    void init(int n)
    {
        pa.clear();
        rank.clear();
        pa.reserve(n);
        rank.reserve(n);
        for (int i = 0; i < n; i++)
        {
            pa[i] = i;
            rank[i] = 1;
        }
    }

    int find(int x)
    {
        return x==pa[x]? x:(pa[x] = find(pa[x]));
    }

    void merge(int i, int j)
    {
        int x = find(i);
        int y = find(j);

        if (rank[y] > rank[x])
        {
            pa[x] = y;
        }
        else
        {
            pa[y] = x;
        }

        if (rank[x] == rank[y] && x != y)
        {
            rank[y]++;
        }
    }

    bool connect(int i, int j)
    {
        return find(i) == find(j);
    }

    int minimumEffortPath(vector<vector<int>>& heights) {
        vector<tuple<int, int, int>> edges;
        int m = heights.size();
        int n = heights[0].size();

        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                int id = i*n + j;
                if (i > 0)
                {
                    edges.emplace_back(std::tuple<int, int, int>(id -n, id, abs(heights[i][j]- heights[i-1][j])));
                }

                if (j > 0)
                {
                     edges.emplace_back(std::tuple<int, int, int>(id -1, id, abs(heights[i][j]- heights[i][j-1])));
                }
            }
        }

        sort(edges.begin(), edges.end(), [&](const auto &e1, const auto &e2){
   
            return std::get<2>(e1) < std::get<2>(e2);
        });

        init(m*n);
        int minCost = 0;
        for (int i = 0; i < edges.size(); i++)
        {
            if (connect(0, m*n-1))
            {
                break;
            }

            merge(std::get<0>(edges[i]), std::get<1>(edges[i]));
            minCost = minCost < std::get<2>(edges[i])? std::get<2>(edges[i]):minCost;
        }

        return minCost;
    }
};

还有一种思路是参考网上的采用