题目描述
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
解题思路
二叉树的层序遍历的非递归需要用到队列
直观理解:将每一层的节点依从丢进队列,节点出队时,将该节点的左孩子和右孩子都丢入队列
1.根结点入队
2.从队列中取出一个元素
3.访问该元素所指结点的左孩子和右孩子,若左孩子和右孩子非空,依次入队
代码逻辑:
1、建立一个队列,将根节点压入队列
2、如果队列非空,用一个变量cnt记录此时队列的长度,即为同一层节点的数目。
3、用一个for循环将同一层的节点出队,该节点出队时判断其左右孩子是否为空,如果非空,将其入队
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
if(root == null)
return res;
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while(!queue.isEmpty()){
List<Integer> level = new ArrayList<>();
//当前所在层的元素总数
int cnt = queue.size();
for(int i = 0; i < cnt; i++){
TreeNode node = queue.poll();
level.add(node.val);
if(node.left != null){
queue.add(node.left);
}
if(node.right != null){
queue.add(node.right);
}
}
res.add(level);
}
return res;
}
}
参考:https://leetcode-cn.com/problems/binary-tree-level-order-traversal/