LeetCode 102. 二叉树的层序遍历

102. 二叉树的层序遍历

给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。

示例:
二叉树:[3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果:

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

解题

因为是层次遍历

如果直接用手算的话,可以得到结果:3、9、20、15、7

我在这篇文章中说明了二叉树遍历算法:
https://blog.csdn.net/qq_17623363/article/details/106612418
其中包括了二叉树的层次遍历。

在层次遍历的时候,我们只需要遍历出来某一层的数据,把这一层的数据加入到List中,然后把这个List加入到最终的list中,最后返回即可

难点在于如何统计。

统计每一层节点的数据:

1、当每一层都入队之后获取这个队列的数量,这个数量就是每一层的节点的数量
2、然后根据这个数量来循遍历出队列中现在存在的每一层的数据
3、循环1和2至此结束

关键代码:

static public List<List<Integer>> levelOrder(TreeNode root) {
       List<List<Integer>> lists = new ArrayList<>();

       //如果直接传入的是空,那就直接返回
       if(root==null)
           return lists;

       //利用队列的特性来存储每层的节点
       LinkedList<TreeNode> queue = new LinkedList<>();

       //默认根节点已经加入到队列
       queue.add(root);



        while(!queue.isEmpty()){

            List<Integer> list = new ArrayList<>();

            int n = queue.size();//记录当前层次的数量

            int index = 0;
            //只要每层遍历的次数符合每层的数量就算是遍历的这同一层的节点。
            while (index<n){
                root = queue.pop();

                //添加到每一层的List中
                list.add(root.val);

                if(root.left!=null) {
                    queue.add(root.left);
                }

                if(root.right!=null) {
                    queue.add(root.right);
                }

                //索引++  遍历同一层次后边的节点
                index++;
            }


            //把每一层的节点加入到最终的List中
            lists.add(list);
        }

        return lists;
    }

全部代码:

package tree.层次遍历二叉树;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Stack;

class Solution {

    public static void main(String[] args) {
        TreeNode t1 = new TreeNode(1);
        TreeNode t2 = new TreeNode(2);
        TreeNode t3 = new TreeNode(3);
        TreeNode t4 = new TreeNode(4);
        TreeNode t5 = new TreeNode(5);
        t1.left=t2;
        t1.right=t3;
        t2.left=t4;
        t2.right=t5;


        System.out.println();
        System.out.println("层次遍历:");
        List<List<Integer>> lists = levelOrder(t1);

        System.out.println("结果:"+lists.toString());

        System.out.println("遍历结果:");

        lists.stream().forEach(li -> {

            li.stream().forEach(ll -> {
                System.out.print(ll.intValue()+" ");
            });

            System.out.println();

        });

    }

   static public List<List<Integer>> levelOrder(TreeNode root) {
       List<List<Integer>> lists = new ArrayList<>();

       //如果直接传入的是空,那就直接返回
       if(root==null)
           return lists;

       //利用队列的特性来存储每层的节点
       LinkedList<TreeNode> queue = new LinkedList<>();

       //默认根节点已经加入到队列
       queue.add(root);



        while(!queue.isEmpty()){

            List<Integer> list = new ArrayList<>();

            int n = queue.size();//记录当前层次的数量

            int index = 0;
            //只要每层遍历的次数符合每层的数量就算是遍历的这同一层的节点。
            while (index<n){
                root = queue.pop();

                //添加到每一层的List中
                list.add(root.val);

                if(root.left!=null) {
                    queue.add(root.left);
                }

                if(root.right!=null) {
                    queue.add(root.right);
                }

                //索引++  遍历同一层次后边的节点
                index++;
            }


            //把每一层的节点加入到最终的List中
            lists.add(list);
        }

        return lists;
    }
}

class TreeNode {
    int val;//每个节点存放的数据
    TreeNode left;//左节点
    TreeNode right;//右节点
    TreeNode(int x) { val = x; }
}

TrueDei CSDN认证博客专家 Linux 分布式 Java
不荒废现在,不畏惧未来!我认为把知识给别人讲会,讲明白,自己才彻底明白。努力把文章写好,写明白每一篇文章,分享给更多人。
©️2020 CSDN 皮肤主题: 程序猿惹谁了 设计师:上身试试 返回首页
实付 49.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值