515. 在每个树行中找最大值

515. 在每个树行中找最大值

题目描述

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

方法一 广度优先遍历（BFS）

二叉树层次遍历，设置一个遍历maxValue记录每一层的最大值。

代码

/**
 * 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) {
        //层次遍历，设置遍历maxValue记录每层的最大值
        vector<int> res;
        if(root == NULL){
            return res;
        }
        queue<TreeNode*> treeQue;
        treeQue.push(root);
        
        while(!treeQue.empty()){
            int sz = treeQue.size();
            TreeNode* node = NULL;
            int maxValue = INT_MIN;
            for(int i = 0;i < sz;++i){
                node = treeQue.front();
                treeQue.pop();
                if(node->val > maxValue){
                    maxValue = node->val;
                }
                if(node->left){
                    treeQue.push(node->left);
                }
                if(node->right){
                    treeQue.push(node->right);
                }
            }
            res.push_back(maxValue);
        }
        return res;
    }
};

方法二 深度优先遍历（DFS）

深度优先遍历二叉树，设置vector更新没一层的最大值。

代码

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