[演算法]按之字形順序列印二叉樹
阿新 • • 發佈:2019-01-10
題目描述
請實現一個函式按照之字形列印二叉樹,即第一行按照從左到右的順序列印,第二層按照從右至左的順序列印,第三行按照從左到右的順序列印,其他行以此類推。
思路
層次遍歷,在此基礎上加上奇數偶數行判斷。
程式碼
import java.util.ArrayList; import java.util.LinkedList; import java.util.Queue; /* public class TreeNode { int val = 0; TreeNode left = null; TreeNode right = null; public TreeNode(int val) { this.val = val; } }*/ public class Solution { private ArrayList<ArrayList<Integer>> listAll = new ArrayList<>(); public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) { if(pRoot == null){ return listAll; } int level = 0; Queue<TreeNode> queue = new LinkedList<>(); queue.offer(pRoot); while(!queue.isEmpty()){ int num = queue.size(); ArrayList<Integer> list = new ArrayList<>(); for(int i = 0; i < num; i++){ TreeNode node = queue.poll();if(level % 2 == 0){ list.add(node.val); }else{ list.add(0, node.val); } if(node.left != null){ queue.offer(node.left); } if(node.right != null){ queue.offer(node.right); } } level++; listAll.add(list); } return listAll; } }