[Leetcode] 515. Find Largest Value in Each Tree Row 解题报告

原创于 2017-12-08 15:18:19 发布 · 182 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode #解题报告 #Tree #DFS

IT公司面试习题专栏收录该内容

736 篇文章

订阅专栏

本文介绍了一种算法，用于找到二叉树每一层的最大值。通过深度优先搜索(DFS)实现，递归地遍历树的各层并记录每层的最大节点值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

You need to find the largest value in each row of a binary tree.

Example:

Input: 

          1
         / \
        3   2
       / \   \  
      5   3   9 

Output: [1, 3, 9]

思路：

还是那句话：遇到树的问题，首先要想到DFS或者BFS。在遍历树的过程中，纪录当前结点所在的层次，然后检查该层的最大值，如果结点的值超过了该层的最大值，那么就更新。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> largestValues(TreeNode* root) {
        vector<int> ret;
        largestValues(root, ret, 0);
        return ret;
    }
private:
    void largestValues(TreeNode *root, vector<int> &ret, int height) {
        if (root == NULL) {
            return;
        }
        if (ret.size() > height) {
            ret[height] = max(ret[height], root->val);
        }
        else {
            ret.push_back(root->val);
        }
        largestValues(root->left, ret, height + 1);
        largestValues(root->right, ret, height + 1);
    }
};