层次遍历Java代码

最新推荐文章于 2023-05-22 22:00:00 发布

有梦想有行动

最新推荐文章于 2023-05-22 22:00:00 发布

阅读量348

点赞数

分类专栏： Java 文章标签： java

本文链接：https://blog.csdn.net/CSDN_of_ding/article/details/116948171

版权

Java 专栏收录该内容

40 篇文章 1 订阅

订阅专栏

层次遍历

public class Solution {
    public int[] levelPrint(TreeNode root){
        if(root == null)
        {
            return new int[0];
        }
        LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
        LinkedList<Integer> arr = new LinkedList<>();
        TreeNode current = null;
        queue.offer(root);//将根节点入队
        while(!queue.isEmpty())
        {
            current = queue.poll();//出队队头元素并访问
            arr.add(current.val);
            //System.out.println(current.val);
            if(current.left != null)//如果当前节点的左节点不为空入队
            {
                queue.offer(current.left);
            }
            if(current.right != null)//如果当前节点的右节点不为空，把右节点入队
            {
                queue.offer(current.right);
            }
        }
        int newarr[] = new int[arr.size()];
        for(int i=0; i<arr.size(); i++){ //#####特别要注意
//            newarr[i]=arr.removeFirst();
            newarr[i]=arr.get(i);
        }
        return newarr;

    }
}