JZ-059-按之字形順序列印二叉樹
阿新 • • 發佈:2022-01-17
按之字形順序列印二叉樹
題目描述
請實現一個函式按照之字形列印二叉樹,即第一行按照從左到右的順序列印,第二層按照從右至左的順序列印,第三行按照從左到右的順序列印,其他行以此類推。
題目連結: 按之字形順序列印二叉樹
程式碼
import java.util.ArrayList; import java.util.LinkedList; import java.util.Queue; import java.util.Stack; /** * 標題:按之字形順序列印二叉樹 * 題目描述 * 請實現一個函式按照之字形列印二叉樹,即第一行按照從左到右的順序列印,第二層按照從右至左的順序列印,第三行按照從左到右的順序列印,其他行以此類推。 * 題目連結: * https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0?tpId=13&&tqId=11212&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking */ public class Jz59 { public ArrayList<ArrayList<Integer>> print(TreeNode pRoot) { Queue<TreeNode> treeNodes = new LinkedList<>(); ArrayList<ArrayList<Integer>> result = new ArrayList<>(); treeNodes.add(pRoot); boolean flag = true; // 當flag為true時,棧從左向右放;當flag為false時,棧從右向左放 while (!treeNodes.isEmpty()) { int cnt = treeNodes.size(); ArrayList<Integer> curList = new ArrayList<>(); Stack<TreeNode> curNodes = new Stack<>(); while (cnt-- > 0) { TreeNode t = treeNodes.poll(); if (t == null) { continue; } curList.add(t.val); if (flag) { curNodes.add(t.left); curNodes.add(t.right); } else { curNodes.add(t.right); curNodes.add(t.left); } } flag = !flag; while (!curNodes.isEmpty()) { treeNodes.add(curNodes.pop()); } if (!curList.isEmpty()) { result.add(curList); } } return result; } public static void main(String[] args) { } }
【每日寄語】 只要你今天再多努力一下,那個未來可以像星星一樣閃閃發光的人就是你呀!