Leetcode 515. Find Largest Value in Each Tree Row

最新推荐文章于 2025-01-12 14:45:00 发布

Z-Pilgrim

最新推荐文章于 2025-01-12 14:45:00 发布

阅读量253

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode题解

本文链接：https://blog.youkuaiyun.com/u011026968/article/details/80932918

LeetCode题解专栏收录该内容

271 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索(DFS)解决LeetCode题目“查找二叉树每层最大值”的方法。通过递归遍历二叉树，记录每一层的最大节点值，最终返回包含所有层最大值的列表。

https://leetcode.com/problems/find-largest-value-in-each-tree-row/description/

跟上一题一样，直接开数组记录就行

class Solution {
public:
    void dfs( TreeNode* root, int cur_dep,  vector<int> &ans) {
        if (!root) return;
        if (cur_dep == ans.size()) {
            ans.push_back( root->val );
        } else {
            ans[cur_dep] = max(ans[cur_dep], root->val);
        }
        dfs(root->left, cur_dep+1, ans);
        dfs(root->right, cur_dep+1, ans);
    }
    
    vector<int> largestValues(TreeNode* root) {
        vector<int> ans;
        dfs(root, 0, ans);
        return ans;
    }
};