【LeetCode515 Find Largest Value in Each Tree Row】二叉树每行的最大值

一、问题描述

给定一个二叉树，二叉树每一行都有一个最大值，将该值保存到一个数组中，最后输出该数组。举例如下：给定二叉树，输出数组[1,3,9]

Input:

1
/ \
3   2
/ \   \
5   3   9

Output: [1, 3, 9]

二、思路详解

这一题是遍历树，给树的每一个节点赋予层数的信息，定义一个数组maxNum数组，存储每层中最大值，其下标i对应于树中的第i层，其中根节点为第0层。在遍历某个树节点时，先判断该节点的层数与当前数组中已存储的数是否相等，若相等，则将该节点值与当前数组最后一位值进行比较，数组取其中的最大值。若不相等，则该节点为该层第一个节点，直接将该节点值赋值给数组对应的位。深度遍历该二叉树，最后输出这个数组即可。

三、代码实现

/**
* 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) {
if(root == NULL) {
return maxNum;
}
travel(root,level);
return maxNum;
}
TreeNode* travel(TreeNode* root, int level) {
if(!root) return NULL;
if(maxNum.size() < level + 1) {
maxNum.push_back(root->val);
} else {
maxNum[level] = max(root->val,maxNum[level]);
}
travel(root->left,level+1);
travel(root->right,level+1);
return root;
}
private:
int level = 0;
vector<int> maxNum;
};