You are a hiker preparing for an upcoming hike. You are given heights
, a 2D array of size rows x columns
, where heights[row][col]
represents the height of cell (row, col)
. You are situated in the top-left cell, (0, 0)
, and you hope to travel to the bottom-right cell, (rows-1, columns-1)
(i.e., 0-indexed). You can move up, down, left, or right, and you wish to find a route that requires the minimum effort.
A route's effort is the maximum absolute difference in heights between two consecutive cells of the route.
Return the minimum effort required to travel from the top-left cell to the bottom-right cell.
Example 1:
Input: heights = [[1,2,2],[3,8,2],[5,3,5]] Output: 2 Explanation: The route of [1,3,5,3,5] has a maximum absolute difference of 2 in consecutive cells. This is better than the route of [1,2,2,2,5], where the maximum absolute difference is 3.
Example 2:
Input: heights = [[1,2,3],[3,8,4],[5,3,5]] Output: 1 Explanation: The route of [1,2,3,4,5] has a maximum absolute difference of 1 in consecutive cells, which is better than route [1,3,5,3,5].
Example 3:
Input: 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]] Output: 0 Explanation: This route does not require any effort.
Constraints:
rows == heights.length
columns == heights[i].length
1 <= rows, columns <= 100
1 <= heights[i][j] <= 106
struct Solution;
use std::cmp::Reverse;
use std::collections::BinaryHeap;
impl Solution {
fn minimum_effort_path(heights: Vec<Vec<i32>>) -> i32 {
let n = heights.len();
let m = heights[0].len();
let mut visited: Vec<Vec<bool>> = vec![vec![false; m]; n];
let mut queue: BinaryHeap<(Reverse<i32>, usize, usize)> = BinaryHeap::new();
let mut res = 0;
queue.push((Reverse(0), 0, 0));
while let Some((Reverse(effort), i, j)) = queue.pop() {
res = res.max(effort);
if i == n - 1 && j == m - 1 {
break;
}
visited[i][j] = true;
if i > 0 && !visited[i - 1][j] {
let diff = heights[i][j] - heights[i - 1][j];
queue.push((Reverse(diff.abs()), i - 1, j));
}
if j > 0 && !visited[i][j - 1] {
let diff = heights[i][j] - heights[i][j - 1];
queue.push((Reverse(diff.abs()), i, j - 1));
}
if i + 1 < n && !visited[i + 1][j] {
let diff = heights[i][j] - heights[i + 1][j];
queue.push((Reverse(diff.abs()), i + 1, j));
}
if j + 1 < m && !visited[i][j + 1] {
let diff = heights[i][j] - heights[i][j + 1];
queue.push((Reverse(diff.abs()), i, j + 1));
}
}
res
}
}
#[test]
fn test() {
let heights = vec_vec_i32![[1, 2, 2], [3, 8, 2], [5, 3, 5]];
let res = 2;
assert_eq!(Solution::minimum_effort_path(heights), res);
let heights = vec_vec_i32![[1, 2, 3], [3, 8, 4], [5, 3, 5]];
let res = 1;
assert_eq!(Solution::minimum_effort_path(heights), res);
let heights = vec_vec_i32![
[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]
];
let res = 0;
assert_eq!(Solution::minimum_effort_path(heights), res);
}