原題地址
https://leetcode.com/problems/pacific-atlantic-water-flow/description/
題意
給定一個數字矩陣眯亦,每個位置的數字表示這個位置水面的高度留拾,高度大于或等于鄰近位置的水面 則 水能從該位置流入鄰近位置甩恼。找出這個矩陣中既能流向太平洋(矩陣的左邊界和上邊界)又能流向大西洋(矩陣的右邊界和下邊界)的位置坐標距帅。
思路
不知道是不是之前都在看動態(tài)規(guī)劃的題燕酷。呆抑〖锌祝看到這題第一個思路就是動態(tài)規(guī)劃 [跪了]
想用動態(tài)規(guī)劃就是和那種找出路徑數目的方法類似办陷,分別對能通往太平洋的和大西洋的做一次说搅,最后找出交集炸枣。但是是有問題的,一個位置能否流向目的地不取決于周圍的位置能否流向目的地。适肠。只要足夠高就可以了
代碼
discuss里的代碼
visited表示能到達目的地的情況霍衫,初值為0,用一個|
操作來更新自己的值侯养。
從矩陣的左邊和上邊出發(fā)敦跌,攜帶值1進行dfs,從右邊和下邊出發(fā)逛揩, 攜帶值2進行dfs柠傍。dfs里的邏輯是,如果當前位置比上一個位置的值更低直接返回(不通)辩稽,否則當前位置的visited值|
上前一個位置的值惧笛。
到最后如果兩個大洋都能到的位置,其visited值應該為01|10=11=3(10進制)
逞泄,找出所有這樣的位置患整,返回即可。
妙啊×3
class Solution {
public:
vector<pair<int, int>> res;
vector<vector<int>> visited;
void dfs(vector<vector<int>>& matrix, int x, int y, int pre, int preval){
if (x < 0 || x >= matrix.size() || y < 0 || y >= matrix[0].size()
|| matrix[x][y] < pre || (visited[x][y] & preval) == preval)
return;
visited[x][y] |= preval;
if (visited[x][y] == 3) res.push_back({x, y});
dfs(matrix, x + 1, y, matrix[x][y], visited[x][y]); dfs(matrix, x - 1, y, matrix[x][y], visited[x][y]);
dfs(matrix, x, y + 1, matrix[x][y], visited[x][y]); dfs(matrix, x, y - 1, matrix[x][y], visited[x][y]);
}
vector<pair<int, int>> pacificAtlantic(vector<vector<int>>& matrix) {
if (matrix.empty()) return res;
int m = matrix.size(), n = matrix[0].size();
visited.resize(m, vector<int>(n, 0));
for (int i = 0; i < m; i++) {
dfs(matrix, i, 0, INT_MIN, 1);
dfs(matrix, i, n - 1, INT_MIN, 2);
}
for (int i = 0; i < n; i++) {
dfs(matrix, 0, i, INT_MIN, 1);
dfs(matrix, m - 1, i, INT_MIN, 2);
}
return res;
}
};
自己的原本的錯誤代碼
class Solution {
public:
vector<pair<int, int>> pacificAtlantic(vector<vector<int>>& matrix) {
vector<pair<int, int>> temp;
int m = matrix.size();
if (m == 0) return temp;
int n = matrix[0].size();
vector<vector<bool>> reachP(m, vector<bool>(n, false));
vector<vector<bool>> reachA(m, vector<bool>(n, false));
vector<pair<int, int>> result;
for (int i = 0; i < m; i++) {
reachP[i][0] = true;
reachA[i][n - 1] = true;
}
for (int j = 0; j < n; j++) {
reachP[0][j] = true;
reachA[m - 1][j] = true;
}
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if ((matrix[i][j] >= matrix[i - 1][j] && reachP[i - 1][j]) || (matrix[i][j] >= matrix[i][j - 1] && reachP[i][j - 1])) {
reachP[i][j] = true;
}
}
}
for (int i = m - 1; i >= 1; i--) {
for (int j = n - 1; j >= 1; j--) {
if (i + 1 < m) {
reachP[i][j] = reachP[i][j] || (reachP[i + 1][j] && matrix[i][j] >= matrix[i + 1][j]);
}
if (j + 1 < m) {
reachP[i][j] = reachP[i][j] || (reachP[i][j + 1] && matrix[i][j] >= matrix[i][j + 1]);
}
}
}
for (int i = m - 2; i >= 0; i--) {
for (int j = n - 2; j >= 0; j--) {
if ((matrix[i][j] >= matrix[i + 1][j] && reachA[i + 1][j]) || (matrix[i][j] >= matrix[i][j + 1] && reachA[i][j + 1])) {
reachA[i][j] = true;
}
}
}
for (int i = 0; i < m - 1; i++) {
for (int j = 0; j < n - 1; j++) {
if (i - 1 >= 0) {
reachA[i][j] = reachA[i][j] || (reachA[i - 1][j] && matrix[i][j] >= matrix[i - 1][j]);
}
if (j - 1 >= 0) {
reachA[i][j] = reachA[i][j] || (reachA[i][j - 1] && matrix[i][j] >= matrix[i][j - 1]);
}
}
}
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (reachA[i][j] && reachP[i][j]) result.push_back(make_pair(i, j));
}
}
return result;
}
};