按之字形程序打印二叉树
题目形容
请实现一个函数依照之字形打印二叉树,即第一行依照从左到右的程序打印,第二层依照从右至左的程序打印,第三行依照从左到右的程序打印,其余行以此类推。
题目链接: 按之字形程序打印二叉树
代码
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) { }}
【每日寄语】 只有你明天再多致力一下,那个将来能够像星星一样闪闪发光的人就是你呀!