层序遍历BFS1.将节点层序输出

分行层序遍历

如果要分层把list输出到res中(也就是一层完了再打印一层),就要在while中加一个for来控制到什么时候停止,一定要注意在加的时候看节点是否为空,是空continue,那怎么知道for循环中的size呢,就是看队列中的节点个数,在开始前的节点个数就是这一层所有的节点个数,全部循环完也就是这一层的下一层节点也全部添加进去了
还要注意如果list长度不为0才能添加到res中,否则最后会多添加一个空list
小技巧是创建一个list,然后每次添加完之后都令list=new ArrayList,这样可以创建几个变量?

 import java.util.ArrayList;
 import java.util.LinkedList;
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res=new ArrayList<List<Integer>>();
        if(root==null)return res;
        Queue<TreeNode> q=new LinkedList<TreeNode>();
        q.offer(root);
        while(!q.isEmpty()){
            int size=q.size();
            List<Integer> list=new ArrayList<Integer>();
            for(int i=0;i<size;i++){
                TreeNode node=q.poll();
                list.add(node.val);
                if(node.left!=null) q.offer(node.left);
                if(node.right!=null) q.offer(node.right);
            }
            res.add(list);
        }
        return res;
    }
}

分行之字层序遍历

首先想到树的层序遍历,用一个队列输出,然后想到逆序输出的方式是用list.add(0,a);或者是用deque,也可以实现双端插入。
还有一种通用的方法是设置两个栈,然后每次输出两行,首先加入一个s1,然后在循环1中向s2加入下一层的节点,此时是先加左后加右,这样等会s2输出的时候就是先输出右边的节点了,然后当s1空的时候。再进入s2的循环中,先输出的就是右边的节点,然后加下一层的节点,这个加的时候要先加右边的节点,这样输出的时候就会先输出左边的节点,正好符合下一层的输出。

class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> res=new ArrayList<>();
        if(root==null)return res;
        Deque<TreeNode> dq=new LinkedList<>();
        dq.offer(root);
        boolean flag=false;
        while(!dq.isEmpty()){
            int size=dq.size();
            Deque<Integer> list=new LinkedList<>();
            for(int i=0;i<size;i++){
                TreeNode node=dq.poll();
                if(flag) list.addFirst(node.val);
                else list.addLast(node.val);
                if(node.left!=null)dq.offer(node.left);
                if(node.right!=null)dq.offer(node.right);
            }
            res.add(new ArrayList(list));
            flag=!flag;          
        } 
        return res;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值