二叉树 | 层序遍历

二叉树的层序遍历

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;//未分配内存