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
整体思路:
层次遍历 从左到右按层输出到一个数组里
两个问题:1.正确输出节点
2.按层输出,每层是一个数组,整体放在一个数组里输出
1.利用一个辅助队列,有节点就存入队列中,利用先入先出的原则,从左往右存入就会从左往右输出。每次从队列中弹出一个节点,记为front。看这个front有没有左孩子,如果有左孩子把左孩子放入到队列中,如果front有右孩子,把右孩子放入到队列中。重复,直到队列为空。
2.根据每次的队列长度我们就知道存入的数组的数是几个(每一层的数组),然后队列为空之前的数组就是输出的大数组
代码:
/*** 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) {vector<vector<int>>ans;//答案if(!root) return ans;//没有节点直接输出queue<TreeNode*>que;//辅助队列que.push(root);//将头节点存入队列while(!que.empty()){//重复寻找左右节点,直到队列为空int size = que.size();//每一次队列的长度//注意:不可以在for循环里直接i < que.size()。因为这样队列的长度只是第一次的不更新ans.push_back(vector<int>());//先存入一个数组(但是没有数)//因为ans.back().push_back()使用的前提是数组不为空for(int i = 0; i < size; i++){//循环遍历当前队列长度TreeNode* front = que.front();que.pop();ans.back().push_back(front -> val);//将这个长度的数组存入数组中//第二次再存入的就是第二个数组,依次类推if(front -> left) que.push(front -> left);//有左节点,在队列中存入左节点if(front -> right) que.push(front -> right);//有右节点,在队列中存入右节点}}return ans;//输出数组}
};