牛客题霸NC15求二叉树的层序遍历Java题解

牛客题霸NC15求二叉树的层序遍历Java题解

https://www.nowcoder.com/practice/04a5560e43e24e9db4595865dc9c63a3?tpId=117&&tqId=34936&rp=1&ru=/ta/job-code-high&qru=/ta/job-code-high/question-ranking

方法：利用队列
解题思路：将节点加入到队列中，利用队列Queue的先进后出，依此弹出节点。如果弹出的节点有左、右子节点，则将左、右子节点加入到队列Queue中，将每一层的节点保存到一个ArrayList中（tmp）,每一层遍历完，将这一层的节点都加入到ArrayList<ArrayList<integer>> list 中。

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>> list = new ArrayList<>(); 
 
        Queue<TreeNode> queue = new LinkedList<>(); 
 
        if(root!=null){           
            queue.add(root);
        }
 
        while(!queue.isEmpty()){
            ArrayList<Integer> tmp = new ArrayList<>();  //存储每一层节点
            for(int i = queue.size();i>0;i--){   //遍历当前层的节点
                TreeNode node = queue.poll();    //弹出队列中的节点
                tmp.add(node.val);               //将此节点加入到当前层的 ArrayList中
                if(node.left!=null){             //如果左子节点不为空，则将其加入到队列中
                    queue.add(node.left);
                }
                if(node.right!=null){       //如果左子节点不为空，则将其加入到队列中
                    queue.add(node.right);
                }
            }
            list.add(tmp);                 //将这一层的节点加入到list中
        }
        return list;
    }
}