LeetCode(树)二叉树的层次遍历

二叉树的层次遍历

给定一个二叉树，返回其按层次遍历的节点值。 （即逐层地，从左到右访问所有节点）。

例如:
给定二叉树: [3,9,20,null,null,15,7],

  3
 / \
9  20
  /  \
 15   7

返回其层次遍历结果：

[
  [3],
  [9,20],
  [15,7]
]

思路：复习了之前写过的层序遍历代码，主要思想是借助一个队列来存储每一层二叉树的节点，然后利用出队列的方式将每一层的节点值依次放入存储Integer的list中，采用了ArrayList来模拟队列。

java代码如下：

/**
 * 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) {
        //使用ArrayList来模拟存储每一层树节点的队列
        ArrayList<TreeNode> treeQueue=new ArrayList<TreeNode>();
        ArrayList<List<Integer>> res=new ArrayList<List<Integer>>();
        
        if(root==null){
            return res;
        }
        treeQueue.add(root);
        
        while(!treeQueue.isEmpty()){
            int len=treeQueue.size();
            ArrayList<Integer> list=new ArrayList<Integer>();
            for(int i=0;i<len;i++){
                 TreeNode temp=treeQueue.remove(0);
                 list.add(temp.val);
                 if(temp.left!=null){
                     treeQueue.add(temp.left);
                 }
                 if(temp.right!=null){
                     treeQueue.add(temp.right);
                 }
            }
            res.add(list);
        }
        return res;
    }
}