【LeetCode-中等】515. 在每个树行中找最大值（C++实现）

最新推荐文章于 2024-08-26 09:06:14 发布

JKL27

最新推荐文章于 2024-08-26 09:06:14 发布

阅读量203

点赞数

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

本文链接：https://blog.csdn.net/m0_52052747/article/details/125581072

版权

LeetCode 专栏收录该内容

24 篇文章 0 订阅

订阅专栏

题目链接

515. 在每个树行中找最大值 - 力扣（LeetCode）

实现思路

使用层序遍历，记录每一行的最大值即可。

q.push(root);
while (!q.empty()) {
    int size = q.size();
    int max = INT_MIN;
    for (int i = 0; i < size; i++) {
        TreeNode * temp = q.front();
        q.pop();
        if (temp->left) {
            q.push(temp->left);
        }
        if (temp->right) {
            q.push(temp->right);
        }
        max = temp->val > max ? temp->val : max;
    }
    ans.push_back(max);
}

实现代码（C++）

class Solution {
public:
    vector<int> largestValues(TreeNode* root) {
        if (root == nullptr) {
            return vector<int>();
        }
        vector<int> ans;

        queue<TreeNode*> q;

        q.push(root);
        while (!q.empty()) {
            int size = q.size();
            int max = INT_MIN;
            for (int i = 0; i < size; i++) {
                TreeNode * temp = q.front();
                q.pop();
                if (temp->left) {
                    q.push(temp->left);
                }
                if (temp->right) {
                    q.push(temp->right);
                }
                max = temp->val > max ? temp->val : max;
            }
            ans.push_back(max);
        }
        
        return ans;
    }
};