(M)DFS:515. Find Largest Value in Each Tree Row

原创于 2017-10-14 14:23:23 发布 · 198 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

264 篇文章

订阅专栏

DFS

35 篇文章

订阅专栏

本文介绍了一种算法，用于求解二叉树中每一层的最大值。通过深度优先搜索遍历二叉树，使用一个变量记录当前深度，并更新对应深度的最大值。

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

这道题的意思是，存下一棵二叉树每一行最大的数。

我的做法是，遍历这棵树，用一个变量depth记录当前的深度，看当前遍历到的这个结点是不是比当前深度下的最大值大，大的话就替换掉当前深度下的最大值。

/**
 * 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> res;
        dfs(root, res, 1);
        return res;
    }
    void dfs(TreeNode* root, vector<int>& res, int depth)
    {
        if(root == NULL)
            return;
        if(res.size() < depth)
        {
            res.push_back(root->val);
        }
        else
        {
            res[depth - 1] = max(res[depth - 1], root->val);
        }
        dfs(root->left, res, depth + 1);
        dfs(root->right, res, depth + 1);
    }
};