LeetCode
矩阵中移动的最大次数
题目链接:2684. 矩阵中移动的最大次数 - 力扣(LeetCode)
题目描述
给你一个下标从 0 开始、大小为 m x n
的矩阵 grid
,矩阵由若干 正 整数组成。
你可以从矩阵第一列中的 任一 单元格出发,按以下方式遍历 grid
:
- 从单元格
(row, col)
可以移动到(row - 1, col + 1)
、(row, col + 1)
和(row + 1, col + 1)
三个单元格中任一满足值 严格 大于当前单元格的单元格。
返回你在矩阵中能够 移动 的 最大 次数。
示例 1:
输入:grid = [[2,4,3,5],[5,4,9,3],[3,4,2,11],[10,9,13,15]]
输出:3
解释:可以从单元格 (0, 0) 开始并且按下面的路径移动:
- (0, 0) -> (0, 1).
- (0, 1) -> (1, 2).
- (1, 2) -> (2, 3).
可以证明这是能够移动的最大次数。
示例 2:
输入:grid = [[3,2,4],[2,1,9],[1,1,7]]
输出:0
解释:从第一列的任一单元格开始都无法移动。
提示:
m == grid.length
n == grid[i].length
2 <= m, n <= 1000
4 <= m * n <= 105
1 <= grid[i][j] <= 106
思路
代码
C++
class Solution {
public:
int maxMoves(vector<vector<int>> &grid) {
int m = grid.size(), n = grid[0].size();
int ans = 0;
function<void(int, int)> dfs = [&](int i, int j) {
ans = max(ans, j);
if (ans == n - 1) { // ans 已达到最大值
return;
}
// 向右上/右/右下走一步
for (int k = max(i - 1, 0); k < min(i + 2, m); k++) {
if (grid[k][j + 1] > grid[i][j]) {
dfs(k, j + 1);
}
}
grid[i][j] = 0;
};
for (int i = 0; i < m; i++) {
dfs(i, 0); // 从第一列的任一单元格出发
}
return ans;
}
};
Java
class Solution {
private int ans;
public int maxMoves(int[][] grid) {
for (int i = 0; i < grid.length; i++) {
dfs(i, 0, grid); // 从第一列的任一单元格出发
}
return ans;
}
private void dfs(int i, int j, int[][] grid) {
ans = Math.max(ans, j);
if (ans == grid[0].length - 1) { // ans 已达到最大值
return;
}
// 向右上/右/右下走一步
for (int k = Math.max(i - 1, 0); k < Math.min(i + 2, grid.length); k++) {
if (grid[k][j + 1] > grid[i][j]) {
dfs(k, j + 1, grid);
}
}
grid[i][j] = 0;
}
}