你有一个用于表示一片土地的整数矩阵land,该矩阵中每个点的值代表对应地点的海拔高度。若值为0则表示水域。由垂直、水平或对角连接的水域为池塘。池塘的大小是指相连接的水域的个数。编写一个方法来计算矩阵中所有池塘的大小,返回值需要从小到大排序。
示例:
输入:
[
[0,2,1,0],
[0,1,0,1],
[1,1,0,1],
[0,1,0,1]
]
输出: [1,2,4]
来源:力扣(LeetCode
题目很正常,思路也简单。
思路:每个点都要循环一遍。如果是0,递归且把走过的路径置为1。
当时有个坑,是不知道计数要怎么计数。
当时在dfs()函数最后一行写到if(cot!=0){ ans.push_back(cot); cot=0; }
答案是错的。第二个样例就不对。
原因:一条路径,两个岔路口(1和2),先走1。走到最终的时候,前面走过的全部置为非零,最终点周围肯定非零,这时我们就开始计数了。但道路2还没走呢。道路1和2应该走完才开始计数。
分析产生的结果:开始考虑计数条件的时候,是这么想的“周围都没有0了,开始计数”,现在想想应该是“这条路走完了开始计数”。
class Solution {
public:
vector<int>ans;
int cot=0;
int x[8]={0,0,1,-1,1,1,-1,-1};
int y[8]={-1,1,0,0,1,-1,-1,1};
vector<int> pondSizes(vector<vector<int>>& land) {
for(int i=0;i<land.size();i++){
for(int j=0;j<land[0].size();j++){
if(land[i][j]==0){
dfs(land,i,j);
if(cot!=0){
ans.push_back(cot);
cot=0;
}
}
}
}
if(!ans.empty())sort(ans.begin(),ans.end());
return ans;
}
void dfs(vector<vector<int>>& land,int px,int py){
if(px<0||py<0||px>=land.size()||py>=land[0].size()||land[px][py]!=0){
return;
}
land[px][py]=1;
cot++;
for(int i=0;i<8;i++) dfs(land,px+x[i],py+y[i]);
}
};