二叉树的层序遍历-Java

题目：二叉树的层序遍历

给定一个二叉树，返回该二叉树层序遍历的结果，（从左到右，一层一层地遍历）
例如：
给定的二叉树是{3,9,20,#,#,15,7}, 该二叉树层序遍历的结果是
[
[3],
[9,20],
[15,7]
]

思路

以前写过层次遍历，在这里。但当时只是system.out.print输出了，并没有按层次输出。
层序遍历需要构造一个队列来操作节点，队列的创建方式别忘了。

---

需要注意的点

• 队列如何创建
• 判root为空时不能返回null，需要返回一个对象。
• 用一个ArrayList类型的数组来存每一层的结果，注意是arr.add(node.val)，而不是arr.add(node)

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @return int整型ArrayList<ArrayList<>>
     */
    public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
        
        ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
        LinkedList<TreeNode> q = new LinkedList<TreeNode>(); 
        q.add(root);
        
        if(root == null){
            return res;
        }
        
        while( !q.isEmpty() ){
            ArrayList<Integer> result = new ArrayList<Integer>();  //result的定义必须在while内部，否则每次result的结果会影响下一次的结果
            int size = q.size();  // 这里必须提前计算q.size，不能将其放在循环条件中（int i = 0; i < q.size(); i++），因为q.size是不断变化的，每次循环都不一样。
            
            for(int i = 0; i < size; i++){  //用for循环来处理每一层的节点
                TreeNode temp = q.poll();
                result.add(temp.val);
                if(temp.left != null){
                    q.add(temp.left);
                }
                if(temp.right != null){
                    q.add(temp.right);
                }
            }
            res.add(result);
        }
        return res;
    }
}