问题:
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
分析:
(1)对树进行层次遍历:使用队列
(2)需要单独记录每一层的序列:两个队列交替使用
code:
//树节点 public class TreeNode { int val = 0; TreeNode left = null; TreeNode right = null; public TreeNode(int val) { this.val = val; } } public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) { /* 层次遍历:队列 奇数行:左——>右 偶数行:右->左(奇数行的逆序) */ ArrayList<ArrayList<Integer>> allList = new ArrayList<ArrayList<Integer>>(); if(pRoot==null){ return null; } Queue<TreeNode>[] queue = new Queue[2]; queue[0] = new ArrayDeque<>();//奇数行 queue[1] = new ArrayDeque<>();//偶数行 int count=1; //记录当前到达的层数 queue[1].add(pRoot); while(!queue[0].isEmpty() || !queue[1].isEmpty()){ ArrayList<Integer> list = new ArrayList<>(); while(!queue[count%2].isEmpty()){ TreeNode temp = queue[count%2].poll(); list.add(temp.val); if(temp.left!=null){ queue[(count+1)%2].add(temp.left); } if(temp.right!=null){ queue[(count+1)%2].add(temp.right); } }
//偶数行:翻转序列 if(count%2==0){ Collections.reverse(list); } allList.add(list); count++; } return allList; }