给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
示例1:
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]
示例 2:
输入:root = [1]
输出:[[1]]
示例 3:
输入:root = []
输出:[]
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/binary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
代码
/**
* 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>> res;//定义一个二维数组保存层序遍历的结果
while(!que.empty())
{
int size = que.size();//size为当前队列的大小
vector<int> vec;
while(size--)
{
TreeNode* node = que.front();
que.pop();
vec.push_back(node->val);//将队头元素放入数组
if(node->left)
que.push(node->left);
if(node->right)
que.push(node->right);
}
//第一遍大循环 根结点6在队中 size=1 一轮循环下来 6出队 size=0,6的左右孩子4,7分别入队
//第二遍大循环 size=2 第一个内循环4出队 4的左右孩子1,3分别入队 size=1 第二个内循环7出队 7的左右孩子5,8入队 size=0 第二层也结束了
//第三遍大循环 size=4 ...
res.push_back(vec);
}
return res;
}
};