题目
You need to find the largest value in each row of a binary tree.
Example:
Input:
1
/ \
3 2
/ \ \
5 3 9
Output: [1, 3, 9]
思路
找到二叉树的每一行中的最大值,然后返回一个vector。
可以很容易想到用层序遍历来实现,遍历每一层保存每一层的最大值。需要用一个值来保存当前层的节点个数,然后将子节点放入一个队列中,从而实现每一层的遍历。
是用一个容器vec来做辅助队列,当vec大小为整个树所有节点数时,说明遍历结束。用一个数cur来辅助这一过程,每遍历一个节点cur加一,并将该节点的左右子节点压入vec中,遍历结束时,树中全部节点都在vec中,而cur为全部节点数。
在遍历每一层时,用i来表示每一层的节点数,用max表示这一层的最大值,访问每个节点时更新max,这一层访问结束后,将max存入结果容器res中。
需要注意的时两层循环之间的关系,以及各自的结束条件。
代码
/**
* 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;
vector<TreeNode*> vec; //辅助队列用于遍历全部节点
vec.push_back(root);
int cur = 0;
int i = 0;
while(cur<vec.size())
{
i = vec.size(); //当前层的节点个数
int max = vec[cur]->val; //用来保存当前层最大值
while(cur<i)
{
if(max < vec[cur]->val)
max = vec[cur]->val;
if(vec[cur]->left != NULL)
vec.push_back(vec[cur]->left);
if(vec[cur]->right != NULL)
vec.push_back(vec[cur]->right);
cur++;
}
res.push_back(max);
}
return res;
}
};