11.二叉树习题

2020-06-08  本文已影响0人  学海一乌鸦

102. 二叉树的层序遍历

在每一层遍历开始前,先记录队列中的结点数量 nn(也就是这一层的结点数量),然后一口气处理完这一层的 n个结点。
最牛B的在于一开始要先计算出来队列的长度: int n=queue.size();
空间复杂度:节点最多不超过n个,最多的为最后一层的叶子节点n/2,故空间复杂度为O(n)
时间复杂度:每个节点进出各一次,所以总体的时间复杂度为O(n)

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> list=new ArrayList<>();
        if(root==null){
            return list;
        }
         Queue<TreeNode> queue=new ArrayDeque<>();
        queue.add(root);
        while(!queue.isEmpty()){
            //一开始预先去queue的大小,后面再加元素也就不用担心了!!!
            int n=queue.size();
            List<Integer> innerList=new ArrayList<>();
            for(int i=0;i<n;i++){
                TreeNode node=queue.poll();
                innerList.add(node.val);
                if(node.left!=null){
                    queue.add(node.left);
                }
                if(node.right!=null){
                    queue.add(node.right);
                }
            }
            list.add(innerList);
        }
        return list;
    }
}
上一篇下一篇

猜你喜欢

热点阅读