?C++刷LeetCode(面试题 16.19. 水域大小)【深度优先搜索】
本帖最后由 糖逗 于 2020-8-3 10:33 编辑题目描述:
面试题 16.19. 水域大小
你有一个用于表示一片土地的整数矩阵land,该矩阵中每个点的值代表对应地点的海拔高度。若值为0则表示水域。由垂直、水平或对角连接的水域为池塘。池塘的大小是指相连接的水域的个数。编写一个方法来计算矩阵中所有池塘的大小,返回值需要从小到大排序。
示例:
输入:
[
,
,
,
]
输出:
提示:
0 < len(land) <= 1000
0 < len(land) <= 1000
class Solution {
private:
vector<vector<int> >direct = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}, {-1, -1}, {-1, 1}, {1, -1}, {1, 1}};
public:
void dfs(vector<vector<int> >&land, int& res1, int i, int j){
if(i < 0 || j < 0 || i >= land.size() || j >= land.size() || land != 0)return;
res1++;
land = 3;
for(auto cha : direct){
dfs(land, res1, i + cha, j + cha);
}
}
vector<int> pondSizes(vector<vector<int>>& land) {
vector<int> res;
int len1 = land.size(), len2 = land.size();
for(int i = 0; i < len1; i++){
for(int j = 0; j < len2; j++){
if(land == 0){
int res1 = 0;
dfs(land, res1, i, j);
res.push_back(res1);
}
}
}
sort(res.begin(), res.end());
return res;
}
}; 下面这种写法为什么会出错?没整明白
class Solution {
private:
vector<vector<int> >direct = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}, {-1, -1}, {-1, 1}, {1, -1}, {1, 1}};
public:
void dfs(vector<vector<int> >&land, int temp, int& res1, int i, int j){
if(i < 0 || j < 0 || i >= land.size() || j >= land.size() || land != 0){
res1 = max(res1, temp);
return;
}
for(auto cha : direct){
land = 3;
dfs(land, temp + 1, res1, i + cha, j + cha);
}
}
vector<int> pondSizes(vector<vector<int>>& land) {
vector<int> res;
int len1 = land.size(), len2 = land.size();
for(int i = 0; i < len1; i++){
for(int j = 0; j < len2; j++){
if(land == 0){
int res1 = 0;
dfs(land, 0, res1, i, j);
res.push_back(res1);
}
}
}
sort(res.begin(), res.end());
return res;
}
};
页:
[1]