1 Star 0 Fork 0

yuhang2__2/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
longest-increasing-path-in-a-matrix.cpp 3.36 KB
一键复制 编辑 原始数据 按行查看 历史
// Time: O(m * n)
// Space: O(m * n)
// topological sort solution
class Solution {
public:
int longestIncreasingPath(vector<vector<int>>& matrix) {
static const vector<pair<int, int>> directions{{0, -1}, {0, 1},
{-1, 0}, {1, 0}};
if (matrix.empty()) {
return 0;
}
vector<vector<int>> in_degree(matrix.size(), vector<int>(matrix[0].size()));
for (int i = 0; i < matrix.size(); ++i) {
for (int j = 0; j < matrix[0].size(); ++j) {
for (const auto& [di, dj] : directions) {
int ni = i + di, nj = j + dj;
if (!(0 <= ni && ni < matrix.size() &&
0 <= nj && nj < matrix[0].size() &&
matrix[ni][nj] > matrix[i][j])) {
continue;
}
++in_degree[i][j];
}
}
}
vector<pair<int, int>> q;
for (int i = 0; i < matrix.size(); ++i) {
for (int j = 0; j < matrix[0].size(); ++j) {
if (!in_degree[i][j]) {
q.emplace_back(i, j);
}
}
}
int result = 0;
while (!q.empty()) {
vector<pair<int, int>> new_q;
for (const auto& [i, j] : q) {
for (const auto& [di, dj] : directions) {
int ni = i + di, nj = j + dj;
if (!(0 <= ni && ni < matrix.size() &&
0 <= nj && nj < matrix[0].size() &&
matrix[i][j] > matrix[ni][nj])) {
continue;
}
if (--in_degree[ni][nj] == 0) {
new_q.emplace_back(ni, nj);
}
}
}
q = move(new_q);
++result;
}
return result;
}
};
// Time: O(m * n)
// Space: O(m * n)
// dfs + memorization solution
class Solution2 {
public:
int longestIncreasingPath(vector<vector<int>>& matrix) {
if (matrix.empty()) {
return 0;
}
int result = 0;
vector<vector<int>> max_lengths(matrix.size(), vector<int>(matrix[0].size()));
for (int i = 0; i < matrix.size(); ++i) {
for (int j = 0; j < matrix[0].size(); ++j) {
result = max(result, longestpath(matrix, i, j, &max_lengths));
}
}
return result;
}
private:
int longestpath(const vector<vector<int>>& matrix, const int i, const int j,
vector<vector<int>> *max_lengths) {
if ((*max_lengths)[i][j] > 0) {
return (*max_lengths)[i][j];
}
int max_depth = 0;
const vector<pair<int, int>> directions{{0, -1}, {0, 1},
{-1, 0}, {1, 0}};
for (const auto& d : directions) {
const int x = i + d.first, y = j + d.second;
if (x >= 0 && x < matrix.size() &&
y >= 0 && y < matrix[0].size() &&
matrix[x][y] < matrix[i][j]) {
max_depth = max(max_depth,
longestpath(matrix, x, y, max_lengths));
}
}
(*max_lengths)[i][j] = max_depth + 1;
return (*max_lengths)[i][j];
}
};
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/yuhang2__2/LeetCode-Solutions.git
git@gitee.com:yuhang2__2/LeetCode-Solutions.git
yuhang2__2
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助