二叉树的层次遍历(递归与迭代)

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

迭代
  • 先将根节点入队列
  • 队列不为空时,要实时更新当前队列长度,节点出队并记录,左右子节点不为空就入队
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {       
        List<List<Integer>> res=new ArrayList<>();  
        if(root==null){return res;}      
        Queue<TreeNode> queue=new LinkedList<>();
        queue.offer(root);
        while(queue.size()>0){  //不为空
            int size=queue.size();  //必须要实时更新队列长度
            List<Integer> list=new LinkedList<>();
            for(int i=0;i<size;i++){
                TreeNode cur=queue.poll();//弹出队头元素,节点出队
                if(cur!=null){
                    list.add(cur.val);  //记录
                    //把每个元素的非空左右子节点进入队列
                    if(cur.left!=null){queue.offer(cur.left);}
                    if(cur.right!=null){queue.offer(cur.right);}
                }
            }
            res.add(list);
        }
        return res;
    }
}

递归
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
         //if(root==null){return new ArrayList<>();}
         List<List<Integer>> res=new ArrayList<>();
         if(root==null){return res;}
         bfs(root,res,1);
         return res;
     }
     public void bfs(TreeNode root,List<List<Integer>> res,int level){
         if(level>res.size()){res.add(new ArrayList<>());}  //新建立一层
         res.get(level-1).add(root.val);  //节点加入对应层
         if(root.left!=null){bfs(root.left,res,level+1);}
         if(root.right!=null){bfs(root.right,res,level+1);}
     }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值