给你一个下标从 0 开始、大小为
m x n的矩阵grid,矩阵由若干 正 整数组成。你可以从矩阵第一列中的 任一 单元格出发,按以下方式遍历
grid:
- 从单元格
(row, col)可以移动到(row - 1, col + 1)、(row, col + 1)和(row + 1, col + 1)三个单元格中任一满足值 严格 大于当前单元格的单元格。返回你在矩阵中能够 移动 的 最大 次数。
cpp
- class Solution {
- public:
- int maxMoves(vector
int >>& grid) { -
- }
- };
python3
- class Solution:
- def maxMoves(self, grid: List[List[int]]) -> int:
我们将第一列的位置入队,然后在网格上进行bfs
对于每个当前位置可以抵达的位置,我们就将其入队,然后记得标记
时间复杂度: O(mn)空间复杂度:O(mn)
- class Solution {
- public:
- typedef pair<int, int> PII;
- static constexpr int dst[3][2] = {{-1, 1}, {0, 1}, {1, 1}};
- int maxMoves(vector
int >>& g) { - queue
q; - int m = g.size(), n = g[0].size(), ret = 0;
- vector
bool>> vis(m, vector<bool>(n)); - for(int i = 0; i < m; i++) q.emplace(i, 0);
- while(q.size()){
- for(int i = 0, ed = q.size(); i < ed; i++){
- auto [x, y] = q.front(); q.pop();
- for(int j = 0, nx, ny; j < 3; j++){
- nx = x + dst[j][0], ny = y + dst[j][1];
- if(nx < 0 || ny < 0 || nx >= m || ny >= n) continue;
- if(vis[nx][ny] || g[x][y] >= g[nx][ny]) continue;
- q.emplace(nx, ny), vis[nx][ny] = 1;
- }
- }
- ret += !q.empty();
- }
- return ret;
- }
- };
- class Solution:
- def maxMoves(self, g: List[List[int]]) -> int:
- m, n = len(g), len(g[0])
- for i in range(m):
- g[i][0] *= -1
- for j in range(n - 1):
- for i, row in enumerate(g):
- if row[j] > 0:
- continue
- for k in i - 1, i, i + 1:
- if 0 <= k < m and g[k][j + 1] > -row[j]:
- g[k][j + 1] *= -1
- if all(row[j + 1] > 0 for row in g):
- return j
- return n - 1