二叉树的层序遍历
1. 题目
给你二叉树的根节点root,返回二叉树的层序遍历序列。
2. 思路:迭代
- 数据结构:队列
- 从根节点开始,先将跟根结点入队
- 依次弹出队列中的结点,将当前结点的左孩子和右孩子分别入队
- 直到遍历到最后一个节点
3. 代码
public static List<List<Integer>> levelOrder(TreeNode root) {//层序遍历
List<List<Integer>> res = new LinkedList<List<Integer>>();
if(root == null) return res;//根节点不为空
Queue<TreeNode> que = new LinkedBlockingQueue<TreeNode>();//当前层的队列
que.add(root);//队列初始化
while(!que.isEmpty()) {//当队列不为空,即上一层有结点
int last_size = que.size();//上一层的个数-que的长度会随着poll()变化,所以要记录
List<Integer> ceng = new LinkedList<Integer>();//标记上一层的结点
for(int i = 0; i < last_size; i++) {//对上一层的每一个节点
TreeNode cur = new TreeNode();//当前操作的节点
cur = que.poll();//上一层的节点还在队列中
ceng.add(cur.val);//加入当前层
if(cur.left != null) {//上一层节点的左结点入队
que.add(cur.left);
}
if(cur.right != null) {//上一层的右结点入队
que.add(cur.right);
}
}//for
res.add(ceng);//把上一层加入结果list
}//while
return res;
}//levelOrder
注意点
que.isEmpty();//分配了内存,但队列中没有值
que == null;//未分配内存