马上注册,结交更多好友,享用更多功能^_^
您需要 登录 才可以下载或查看,没有账号?立即注册
x
本帖最后由 糖逗 于 2020-8-3 10:33 编辑
题目描述:面试题 16.19. 水域大小
你有一个用于表示一片土地的整数矩阵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
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[0].size() || land[i][j] != 0)return;
res1++;
land[i][j] = 3;
for(auto cha : direct){
dfs(land, res1, i + cha[0], j + cha[1]);
}
}
vector<int> pondSizes(vector<vector<int>>& land) {
vector<int> res;
int len1 = land.size(), len2 = land[0].size();
for(int i = 0; i < len1; i++){
for(int j = 0; j < len2; j++){
if(land[i][j] == 0){
int res1 = 0;
dfs(land, res1, i, j);
res.push_back(res1);
}
}
}
sort(res.begin(), res.end());
return res;
}
};
|