题目:
给定一棵二叉树的根节点 root ,请找出该二叉树中每一层的最大值。
示例:
输入: root = [1,3,2,5,3,null,9]
输出: [1,3,9]
思路:采用层序遍历方式,保留每一层的最大值即可。
首先明确:层序遍历“前三式”
- 定义一个队列存放结点指针;
- 定义一个一维数组或二维数组(根据函数返回情况而定);
- 判断函数输入参数root指针是否非空,只有非空才将根节点插入队列。
然后进入while(!que.empty())循环,程序在que非空的情况下运行。
有一个很重要的点在于,每层的节点数是确定的,所以我们在遍历每层结点之前要先获取节点数:int size=que.size();一定放在for()循环之前。
后续进行取值比较,访问结点之后将该节点指针从que里弹出。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<int> largestValues(TreeNode* root) {
queue<TreeNode*> que;
if(root!=nullptr) que.push(root);
vector<int> res;
while(!que.empty()){
int max=INT_MIN;
int size=que.size();
for(int i=0;i<size;i++){
TreeNode* cur=que.front();
max=max>cur->val?max:cur->val;
que.pop();
if(cur->left) que.push(cur->left);//非空左结点入队列
if(cur->right) que.push(cur->right);//非空右结点入队列
}
res.push_back(max);//因为是每层的最大值,所以max在每层遍历之后放进数组中
}
return res;
}
};