按之字形打印二叉树

题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
思路:两个栈,一个栈存储一层的节点,根据栈先进后出的特点,可以让每层打印的节点与放入的节点顺序相反。以此来达到每层“逆序”的特点。

public ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
        
        ArrayList<ArrayList<Integer>> lists = new ArrayList<ArrayList<Integer>>();
        if(pRoot == null){
            return lists;
        }
        
        //Queue<TreeNode> queue1 = new LinkedList<TreeNode>();
        Stack<TreeNode> stack1 =  new Stack<TreeNode>();
         Stack<TreeNode> stack2 =  new Stack<TreeNode>();

        stack1.push(pRoot);
        while(!stack1.isEmpty()||!stack2.isEmpty()){
             ArrayList<Integer> list1 = new ArrayList<Integer>();
            while(!stack1.isEmpty()){
                 TreeNode node = stack1.pop();
                if(node.left!=null){
                    stack2.push(node.left);
                }
                if(node.right!=null){
                    stack2.push(node.right);
                }
                list1.add(node.val);
            }
            if(list1.size()!=0){
                  lists.add(list1);
            }
           
            ArrayList<Integer> list2 = new ArrayList<Integer>();
            while(!stack2.isEmpty()){
               TreeNode node =  stack2.pop();
                if(node.right!=null){
                    stack1.push(node.right);
                }
                if(node.left!=null){
                    stack1.push(node.left);
                }
                list2.add(node.val);
            }
            if(list2.size()!=0){
                 lists.add(list2);
            }
        }
        return lists;
    }

还有一道变形题,每层节点按从左到右的顺序打印节点
题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
思路:
利用队列的先进先出的思想

ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        
   
        ArrayList<ArrayList<Integer> > lists = new ArrayList<ArrayList<Integer> >();
          if(pRoot==null){
            return lists;
            }
        Queue<TreeNode> queue1 = new LinkedList<TreeNode>();
        Queue<TreeNode> queue2 = new LinkedList<TreeNode>();
        queue1.offer(pRoot);
        
        while(!queue1.isEmpty()||!queue2.isEmpty()){
            ArrayList<Integer> list1 = new ArrayList<Integer>();
            while(!queue1.isEmpty()){
              TreeNode node =   queue1.poll();
                list1.add(node.val);
                if(node.left!=null){
                    queue2.offer(node.left);
                }
                if(node.right!=null){
                    queue2.offer(node.right);
                }
            }
            if(list1.size()!=0){
                lists.add(list1);
            }
            ArrayList<Integer> list2 = new ArrayList<Integer>();
            while(!queue2.isEmpty()){
               TreeNode node =  queue2.poll();
                list2.add(node.val);
                 if(node.left!=null){
                    queue1.offer(node.left);
                }
                if(node.right!=null){
                    queue1.offer(node.right);
                }
            }
            if(list2.size()!=0){
                  lists.add(list2);
            }
        }
        return lists;
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值