剑指 Offer II 044. 二叉树每层的最大值

最新推荐文章于 2024-10-31 16:16:13 发布

RIOTSAKURA

最新推荐文章于 2024-10-31 16:16:13 发布

阅读量295

点赞数

分类专栏：刷题之路文章标签：深度优先 leetcode 算法

本文链接：https://blog.csdn.net/m0_50012276/article/details/123127354

版权

刷题之路专栏收录该内容

46 篇文章 0 订阅

订阅专栏

如果已经做过二叉树的层序遍历这题，那么这题其实也是差不多的思路，我们通过一个BFS，在过程中记录下每一层的最大值，并将其记录保存下来即可。AC代码如下：

class Solution {
public:
    vector<int> largestValues(TreeNode* root) {
        vector<int> ans;
        if(root == nullptr) {
            return ans;
        }
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty()) {
            int maxx = que.front()->val;
            int n = que.size();
            for(int i=0;i<n;i++) {
                TreeNode* node = que.front();
                maxx = max(maxx, node->val);
                que.pop();
                if(node->left != nullptr) {
                    que.push(node->left);
                }
                if(node->right != nullptr) {
                    que.push(node->right);
                }
            }
            ans.push_back(maxx);
        }
        return ans;
    }
};