LeetCode第515题解析

最新推荐文章于 2022-07-30 09:47:08 发布

iblue_coffee

最新推荐文章于 2022-07-30 09:47:08 发布

阅读量170

点赞数

分类专栏： LeetCode 二叉树

本文链接：https://blog.csdn.net/blue_coffeei/article/details/107736619

版权

LeetCode 同时被 2 个专栏收录

94 篇文章 3 订阅

订阅专栏

二叉树

14 篇文章 0 订阅

订阅专栏

您需要在二叉树的每一行中找到最大的值。

示例：

输入: 

          1
         / \
        3   2
       / \   \  
      5   3   9 

输出: [1, 3, 9]

class Solution {
public:
    //DFS
    vector<int> largestValues(TreeNode* root) {
        vector<int> res;
        queue<TreeNode*> que;
        if(root) que.push(root);
        while(!que.empty()) {
            int length = que.size();
            int maxValue = INT_MIN;
            for(int i = 0; i < length; ++i) {
                TreeNode* tmp = que.front();
                que.pop();
                maxValue = max(maxValue, tmp->val);
                if(tmp->left) que.push(tmp->left);
                if(tmp->right) que.push(tmp->right);
            }
            res.push_back(maxValue);
        }
        return res;
    }
};

class Solution {
public:
    vector<int> largestValues(TreeNode* root) {
        vector<int> res;
        bfs(0, root, res);
        return res;
    }

    void bfs(int level, TreeNode* root, vector<int>& res) {
        if(root) {
            if(level >= res.size()) res.push_back(root->val);
            else res[level] = max(res[level], root->val);
            res[level] = max(res[level], root->val);
            bfs(level + 1, root->left, res);
            bfs(level + 1, root->right, res);
        }
    }
};