二叉树的层序遍历

  1. 广度优先搜索(层序遍历)迭代法,用队列
  2. 题目

d9d2047b14554aae80d08e9e4236e4be.png

 思路:用队列,num统计每一层的结点数。队列不为空为循环条件。如果num不为0,出队列,并把它的左右孩子入队列。num为0,表示一层遍历完成,将这一层添加到resul中。(resul的定义的意思是往resul中添加List),当想清除list不能用clear,因为resul中存的list的地址,如果给list清空,resul中数据也会清空。但是可以初始化list,得到新的地址再存下一层结点。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        //层序遍历(队列)
        //num记录每一层的结点个数
        Queue<TreeNode>queue=new LinkedList<TreeNode>();
         ArrayList<Integer>list=new ArrayList<>();
        List<List<Integer>>resul=new ArrayList<>();
         int num=0;
         TreeNode node;
         //头结点不为空,添加头结点
         if(root!=null){
           queue.add(root);
           num=queue.size();
         }else{
             //为空则返回resul
             return resul;}
         while(!queue.isEmpty())
         {
             //出栈,入栈它的左右孩子
             if(num!=0) {
                 //出栈
                 node=queue.poll();
                 //存入数组
                 list.add(node.val);
                 //存左结点
                 if(node.left!=null)
                 queue.add(node.left);
                 //存右结点
                 if(node.right!=null)
                 queue.add(node.right);
                 num--;
             }else {
                 //一层的数添加到结果数组
                 resul.add(list);
                 //不能用list.clear(),因为resul里存的list的地址
                 //如果用该方法,list就为空,resul也会为空
                 // 这里我们初始化list,它会指向新的对象,因此地址也会不一样
                 list=new ArrayList<>();
                 num=queue.size();
             }
         }
         //最后一层的节点因为没有孩子,当它们都存入list中
         //由于栈为空,没有进行else里面的添加操作,因此要自己补上
          resul.add(list);
         return resul;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值