面试题 16.19.水域大小

你有一个用于表示一片土地的整数矩阵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]);
        
    }
};