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

Python ml

已于 2022-09-30 12:50:46 修改

阅读量103

点赞数

分类专栏： LeetCode 文章标签：算法

于 2022-09-29 23:35:26 首次发布

本文链接：https://blog.csdn.net/m0_48711099/article/details/127115841

版权

LeetCode 专栏收录该内容

165 篇文章 0 订阅

订阅专栏

class Solution {
public:
    vector<int> largestValues(TreeNode* root) {
        vector<int> ans;
        if(!root) return ans;
        queue<TreeNode*> Qnode;
        Qnode.push(root);
        while (!Qnode.empty()){
            int size = Qnode.size();
            int layermax = INT_MIN;
            while(size--){
                TreeNode* node = Qnode.front();
                Qnode.pop();
                if(node->val>layermax) layermax=node->val;
                if(node->left) Qnode.push(node->left);
                if(node->right) Qnode.push(node->right);
            }
            ans.emplace_back(layermax);
        }
        return ans;
    }
};

dfs (空间复杂度O(height))

class Solution {
public:
    void dfs(vector<int>&res,TreeNode*node,int curHeight){
        if(curHeight==res.size()) {    //第一次遍历到curHeight，扩充res
            res.push_back(node->val);
        }else{
            res[curHeight] = max(res[curHeight],node->val);
        }
        if(node->left){
            dfs(res,node->left,curHeight+1);
        }
        if(node->right){
            dfs(res,node->right,curHeight+1);
        }
    }
    vector<int> largestValues(TreeNode* root) {
        vector<int> res;
        if(!root) return res;
        dfs(res,root,0);
        return res;
    }
};