鱼C论坛

 找回密码
 立即注册
查看: 1742|回复: 1

[技术交流] ?C++刷LeetCode(面试题 16.19. 水域大小)【深度优先搜索】

[复制链接]
发表于 2020-8-3 10:30:42 | 显示全部楼层 |阅读模式

马上注册,结交更多好友,享用更多功能^_^

您需要 登录 才可以下载或查看,没有账号?立即注册

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

本帖被以下淘专辑推荐:

想知道小甲鱼最近在做啥?请访问 -> ilovefishc.com
回复

使用道具 举报

 楼主| 发表于 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[0].size() || land[i][j] != 0){
            res1 = max(res1, temp);
            return;
        }
        for(auto cha : direct){
            land[i][j] = 3;
            dfs(land, temp + 1, 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, 0, res1, i, j);
                    res.push_back(res1);
                }
            }
        }
        sort(res.begin(), res.end());
        return res;
    }
};
想知道小甲鱼最近在做啥?请访问 -> ilovefishc.com
回复 支持 反对

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

小黑屋|手机版|Archiver|鱼C工作室 ( 粤ICP备18085999号-1 | 粤公网安备 44051102000585号)

GMT+8, 2025-1-13 10:25

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

快速回复 返回顶部 返回列表