按照之字形打印二叉树

这里写图片描述

import java.util.*;

 class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}

public class Solution {
    //之字形打印二叉树
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
          ArrayList<ArrayList<Integer>> re=new ArrayList<ArrayList<Integer>>();
          if(pRoot==null)
               return re;
          ArrayList<Integer>list=new ArrayList<>();
          LinkedList<TreeNode>queue=new LinkedList<>();
          queue.offer(null);
          queue.offer(pRoot);
          boolean leftToright=true;  //从左向右打印
          while(queue.size()!=1)
          {
             TreeNode temp=queue.poll();
             if(temp==null) //为分层分隔符
             { 
                Iterator<TreeNode>iterator=null;
                if(leftToright)
                {
                     iterator=queue.iterator(); //正向迭代器
                }else{
                     iterator=queue.descendingIterator(); //反向迭代器
                }
                leftToright=!leftToright; //颠倒方向
                while(iterator.hasNext()){

                      TreeNode node=(TreeNode)iterator.next();
                      list.add(node.val);
                }
                re.add(new ArrayList<Integer>(list));
                list.clear();
                queue.offer(null); //添加分隔符
                continue;
             }
            if(temp.left!=null)
            {
               queue.offer(temp.left);
            }
            if(temp.right!=null)
            {
               queue.offer(temp.right);
            }

          }

          return re;

    }


   //二叉树的层序遍历
    public void TreeBFS(TreeNode pRoot)
    {
     Queue<TreeNode> arr=new LinkedList<TreeNode>();//存放元素的队列
     arr.offer(pRoot);
     boolean flag=false;
     while(!arr.isEmpty())
     {
           TreeNode temp=arr.poll();
           System.out.println(temp.val);
            if(temp.left!=null)
            {
               arr.offer(temp.left);
            }
            if(temp.right!=null)
            {
               arr.offer(temp.right);
            }

     }

    }
    public static void main(String[]args){
         //System.out.println("Hello");
        TreeNode root=new TreeNode(1);
        root.left=new TreeNode(2);
        root.right=new TreeNode(3);
        root.left.left=new TreeNode(4);
        root.left.right=new TreeNode(5);
        root.right.left=new TreeNode(6);
        root.right.right=new TreeNode(7);
        Solution s=new Solution();
        s.TreeBFS(root);
        ArrayList<ArrayList<Integer>>arr=s.Print(root);
        for(ArrayList<Integer>arr2:arr)
        {
             for(int k:arr2)
             {
                 System.out.print(k+" ");
             }
             System.out.println();
        }

    }

}

这里写图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值