你有一个用于表示一片土地的整数矩阵land,该矩阵中每个点的值代表对应地点的海拔高度。若值为0则表示水域。由垂直、水平或对角连接的水域为池塘。池塘的大小是指相连接的水域的个数。编写一个方法来计算矩阵中所有池塘的大小,返回值需要从小到大排序。
示例:
输入:
[
[0,2,1,0],
[0,1,0,1],
[1,1,0,1],
[0,1,0,1]
]
输出: [1,2,4]
提示:
0 < len(land) <= 1000
0 < len(land[i]) <= 1000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pond-sizes-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:递归,搜索,8个方向的深搜
class Solution {
public:
vector<int> pondSizes(vector<vector<int>>& land) {
vector<int> ponds;
// 遍历每一个池塘中的格子
for (int i=0; i<land.size(); ++i) {
for (int j=0;j<land[i].size(); ++j) {
if (land[i][j] == 0) { // 如果发现一个为0的格子,即发现一块水域,则以此为圆心开始进行递归搜索
int size = dfs(land, land.size(), land[0].size(), i, j);
ponds.push_back(size);
}
}
}
sort(ponds.begin(), ponds.end());
return ponds;
}
// land为池塘,row为池塘的行数,col为池塘的列数,i j 为当前所处的池塘位置
int dfs(vector<vector<int>> & land,int row, int col, int i, int j) {
// 边界处理,如果超出了池塘的边界,返回0
if (i < 0 || j < 0 || i >= row || j >= col) {
return 0;
}
if (land[i][j] == 0) { // 如果当前位置是水域,则以此为圆心,周围8个方向为路径,进行深度优先搜索
land[i][j] = 1; // 每经过一块区域,需要对此处进行标记,表示已经来过,避免重复计算
// 返回池塘的大小,大小为当前的1 加上其他八个方向的大小
return 1 + dfs(land, row, col, i-1, j-1) +
dfs(land, row, col, i-1, j) +
dfs(land, row, col, i-1, j+1) +
dfs(land, row, col, i, j-1) +
dfs(land, row, col, i, j+1) +
dfs(land, row, col, i+1, j-1) +
dfs(land, row, col, i+1, j) +
dfs(land, row, col, i+1, j+1);
}
else { // 当前位置不是水域,返回0,表示此路不通,停止搜索
return 0;
}
}
};
执行结果:
通过
显示详情
执行用时:264 ms, 在所有 C++ 提交中击败了92.48%的用户
内存消耗:31.6 MB, 在所有 C++ 提交中击败了52.13%的用户
(妹想到运行还挺快的)