糖逗 发表于 2020-8-3 10:30:42

?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;
    }
};

糖逗 发表于 2020-8-3 10:31:38

下面这种写法为什么会出错?没整明白
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]
查看完整版本: ?C++刷LeetCode(面试题 16.19. 水域大小)【深度优先搜索】