Java教程

lc102-二叉树层序遍历-java实现

本文主要是介绍lc102-二叉树层序遍历-java实现,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

思路:手动构建res结果集合,以及队列;
将当前树节点入队;
将已进入本队列的元素取出,再放入临时集合path中;
按照层数添加叶子节点进入队列;
循环以跳出当前层后将path放入res结果集中。

需要注意的是:
队列的尺寸是随着访问的树的层数而变化的,所以队列尺寸必须在内层循环之外设置为变量,再作为循环退出的条件。

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res =new ArrayList<>();
        Queue<TreeNode> queue =new ArrayDeque<>();
        if(root!=null) queue.add(root);   //判空操作遗忘通不过测试用例!!!
        while(!queue.isEmpty()){
            List<Integer> path =new ArrayList<>();
            int n =queue.size();
            for(int i =0 ;i<n;i++){
                TreeNode node =queue.poll();
                path.add(node.val);
                if(node.left!=null) queue.add(node.left);
                if(node.right!=null) queue.add(node.right);
            }
            res.add(path);
        }
        return res;
    }
}
这篇关于lc102-二叉树层序遍历-java实现的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!