二叉树层序遍历
在本文中,我将会选取LeetCode上二叉树层序遍历的多道例题,并给出解答,通过多道题我们就可以发现,二叉树的层序遍历并不复杂,并且有着共通点。
102. 二叉树的层序遍历
给你二叉树的根节点 root
,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]
示例 2:
输入:root = [1]
输出:[[1]]
示例 3:
输入:root = []
输出:[]
提示:
- 树中节点数目在范围
[0, 2000]
内 -1000 <= Node.val <= 1000
思考:
本题又是一种新的二叉树遍历方式,层序遍历,这种遍历方式其实较为简单,符合正常的遍历顺序。所以我们可以采用队列这种数据结构,运用其先进先出的特性进行遍历。
利用队列遍历:
本题要求按每一层遍历,所以我们采用for循环的形式,进行每一层的遍历。值得注意的是,随着结点加入队列中,队列的大小也会随之改变,所以在进行for循环之前,需要提前存储队列的大小size。
/**
* 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<vector<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> que;
if (root != NULL) que.push(root);
vector<vector<int>> result;
while (!que.empty())
{
vector<int> childvector;//子容器,储存每一层遍历的结果
int size = que.size();//记录队列当前大小
for (int i = 0; i < size; i++)
{
TreeNode* cur = que.front();
que.pop();
if (cur->left)
{
que.push(cur->left);
}
if (cur->right)
{
que.push(cur->right);
}
childvector.push_back(cur->val);
}
result.push_back(childvector);
}
return result;
}
};
递归法:
当然,该题也可以使用递归法来实现,该题中,递归的终止条件为:遍历到空节点,此时说明已将所有结点遍历完毕,进行返回,结束递归。每轮递归中,利用depth记录层数,当进入新的一层时,创建新的子容器。
//递归法
class Solution {
public:
void order(TreeNode* cur, vector<vector<int>>& result, int depth)
{
if (cur == nullptr) return;
if (result.size() == depth) result.push_back(vector<int>());//添加子容器
result[depth].push_back(cur->val);
order(cur->left, result, depth + 1);
order(cur->right, result, depth + 1);
}
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> result;
int depth = 0;
order(root, result, depth);
return result;
}
};
107. 二叉树的层序遍历 II
给你二叉树的根节点 root
,返回其节点值 自底向上的层序遍历 。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:[[15,7],[9,20],[3]]
示例 2:
输入:root = [1]
输出:[[1]]
示例 3:
输入:root = []
输出:[]
思考:
本题与102. 二叉树的层序遍历类似,只需在102题的基础上,对最后的result数组进行翻转操作即可。
代码:
#include <iostream>
#include<vector>
#include<queue>
using namespace std;
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<vector<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> que;
if (root != NULL) que.push(root);
vector<vector<int>> result;
while (!que.empty())
{
vector<int> childvector;//子容器,储存每一层遍历的结果
int size = que.size();//记录队列当前大小
for (int i = 0; i < size; i++)
{
TreeNode* cur = que.front();
que.pop();
if (cur->left)
{
que.push(cur->left);
}
if (cur->right)
{
que.push(cur->right);
}
childvector.push_back(cur->val);
}
result.push_back(childvector);
}
return result;
}
};
199. 二叉树的右视图
给定一个二叉树的 根节点 root
,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例 1:
输入: [1,2,3,null,5,null,4]
输出: [1,3,4]
示例 2:
输入: [1,null,3]
输出: [1,3]
示例 3:
输入: []
输出: []
思考:
本题仍然是一道经典的层序遍历二叉树的题,题意要求只输出最右侧的结点,所以我们只需要在每层for循环中加以判断,只输出目标结点即可。
代码:
#include <iostream>
#include<vector>
#include<queue>
using namespace std;
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> rightSideView(TreeNode* root) {
queue<TreeNode*> que;
if (root != NULL) que.push(root);
vector<int> result;
while (!que.empty())
{
vector<int> childvector;//子容器,储存每一层遍历的结果
int size = que.size();//记录队列当前大小
for (int i = 0; i < size; i++)
{
TreeNode* cur = que.front();
que.pop();
if (cur->left)
{
que.push(cur->left);
}
if (cur->right)
{
que.push(cur->right);
}
if (i == (size - 1))
{
result.push_back(cur->val);//只输出最后一个结点
}
}
}
return result;
}
};
637. 二叉树的层平均值
给定一个非空二叉树的根节点 root
, 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5
以内的答案可以被接受。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:[3.00000,14.50000,11.00000]
解释:第 0 层的平均值为 3,第 1 层的平均值为 14.5,第 2 层的平均值为 11 。
因此返回 [3, 14.5, 11] 。
示例 2:
输入:root = [3,9,20,15,7]
输出:[3.00000,14.50000,11.00000]
提示:
- 树中节点数量在 [1, 104] 范围内
- -231 <= Node.val <= 231 - 1
思考:
本题的本质其实还是二叉树的层序遍历,区别在于在遍历完成后,要求取每一层的平均值,并进行输出。
代码:
值得注意的是,这里需要返回的值时double类型的。
#include <iostream>
#include<vector>
#include<queue>
using namespace std;
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<double> averageOfLevels(TreeNode* root) {
queue<TreeNode*> que;
if (root != NULL) que.push(root);
vector<double> result;
while (!que.empty())
{
double sum = 0;//记录每一层的和
int size = que.size();//记录队列当前大小
for (int i = 0; i < size; i++)
{
TreeNode* cur = que.front();
que.pop();
if (cur->left)
{
que.push(cur->left);
}
if (cur->right)
{
que.push(cur->right);
}
sum += cur->val;
}
result.push_back(sum/size);
}
return result;
}
};
429. N 叉树的层序遍历
给定一个 N 叉树,返回其节点值的层序遍历。(即从左到右,逐层遍历)。
树的序列化输入是用层序遍历,每组子节点都由 null 值分隔(参见示例)。
示例 1:
输入:root = [1,null,3,2,4,null,5,6]
输出:[[1],[3,2,4],[5,6]]
示例 2:
输入:root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出:[[1],[2,3,4,5],[6,7,8,9,10],[11,12,13],[14]]
提示:
- 树的高度不会超过
1000
- 树的节点总数在
[0, 10^4]
之间
思考:
本题又是一道层序遍历题,但这一次的树不再是二叉树,而是一棵N叉树
,每个节点都有可能有多个儿子,而不是像二叉树只含有左右结点,这就要求我们呢在遍历时做出一些调整。
代码:
在进行子儿子遍历时,再采取一层for循环,能够完成完整的遍历。
#include <iostream>
#include<vector>
#include<queue>
using namespace std;
class Node {
public:
int val;
vector<Node*> children;
Node() {}
Node(int _val) {
val = _val;
}
Node(int _val, vector<Node*> _children) {
val = _val;
children = _children;
}
};
class Solution {
public:
vector<vector<int>> levelOrder(Node* root) {
queue<Node*> que;
if (root != NULL) que.push(root);
vector<vector<int>> result;
while (!que.empty())
{
vector<int> childvector;//子容器,储存每一层遍历的结果
int size = que.size();//记录队列当前大小
for (int i = 0; i < size; i++)
{
Node* cur = que.front();
que.pop();
for (int i = 0; i < cur->children.size(); i++)
{
if(cur->children[i])
{
que.push(cur->children[i]);
}
}
childvector.push_back(cur->val);
}
result.push_back(childvector);
}
return result;
}
};
515. 在每个树行中找最大值
给定一棵二叉树的根节点 root
,请找出该二叉树中每一层的最大值。
示例1:
输入: root = [1,3,2,5,3,null,9]
输出: [1,3,9]
示例2:
输入: root = [1,2,3]
输出: [1,3]
提示:
- 二叉树的节点个数的范围是
[0,104]
- -231 <= Node.val <= 231 - 1
思考
本题所考察的仍然是二叉树的层序遍历,只需要在每一层遍历时找出最大值即可。
代码:
#include <iostream>
#include<vector>
#include<queue>
using namespace std;
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 != NULL) que.push(root);
vector<int> result;
while (!que.empty())
{
int size = que.size();//记录队列当前大小
int max= INT_MIN;//初始化每一层的最大值
for (int i = 0; i < size; i++)
{
TreeNode* cur = que.front();
que.pop();
if (cur->left)
{
que.push(cur->left);
}
if (cur->right)
{
que.push(cur->right);
}
max = cur->val > max ? cur->val:max;//更新最大值
}
result.push_back(max);
}
return result;
}
};
116. 填充每个节点的下一个右侧节点指针
给定一个 完美二叉树 ,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL
。
初始状态下,所有 next 指针都被设置为 NULL
。
示例 1:
输入:root = [1,2,3,4,5,6,7]
输出:[1,#,2,3,#,4,5,6,7,#]
解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。序列化的输出按层序遍历排列,同一层节点由 next 指针连接,'#' 标志着每一层的结束。
示例 2:
输入:root = []
输出:[]
提示:
- 树中节点的数量在 [0, 212 - 1] 范围内
-1000 <= node.val <= 1000
思考:
本题的题意比较难懂,其实就是在每一层的层次遍历时记录下本层的头结点,并且在遍历时让前一个结点指向该结点即可。
代码:
class Solution {
public:
Node* connect(Node* root) {
queue<Node*> que;
if (root != NULL) que.push(root);
while (!que.empty()) {
int size = que.size();
// vector<int> vec;
Node* nodePre;
Node* node;
for (int i = 0; i < size; i++) {
if (i == 0) {
nodePre = que.front(); // 取出一层的头结点
que.pop();
node = nodePre;
}
else {
node = que.front();
que.pop();
nodePre->next = node; // 本层前一个节点next指向本节点
nodePre = nodePre->next;
}
if (node->left) que.push(node->left);
if (node->right) que.push(node->right);
}
nodePre->next = NULL; // 本层最后一个节点指向NULL
}
return root;
}
};
(117题也完全可以使用以上代码,以上的求解过程与是否为完美二叉树并没有关系)
参考:代码随想录
往期回顾:
LeetCode102. 二叉树的层序遍历
LeetCode144、145、94. 二叉树遍历
LeetCode18. 四数之和
LeetCode15. 三数之和
LeetCode383. 赎金信
LeetCode454. 四数相加 II
LeetCode1. 两数之和
LeetCode202. 快乐数
LeetCode350. 两个数组的交集 II
LeetCode349. 两个数组的交集
LeetCode1002. 查找共用字符