请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,即第一行按照从左到右的顺序打印,第二层按照从右到左顺序打印,第三行再按照从左到右的顺序打印,其他以此类推。
按之字形顺序打印二叉树需要两个栈。我们在打印某一行结点时,把下一层的子结点保存到相应的栈里。如果当前打印的是奇数层,则先保存左子结点再保存右子结点到一个栈里;如果当前打印的是偶数层,则先保存右子结点再保存左子结点到第二个栈里。
public class Test { private static class BinaryTreeNode { private int val; private BinaryTreeNode left; private BinaryTreeNode right; public BinaryTreeNode() { } public BinaryTreeNode(int val) { this.val = val; } @Override public String toString() { return val + ""; } } public static void print(BinaryTreeNode root) { if (root == null) { return; } List<BinaryTreeNode> current = new LinkedList<>(); List<BinaryTreeNode> reverse = new LinkedList<>(); int flag = 0; BinaryTreeNode node; current.add(root); while (current.size() > 0) { // 从最后一个开始取 node = current.remove(current.size() - 1); System.out.printf("%-3d", node.val); // 当前是从左往右打印的,那就按从左往右入栈 if (flag == 0) { if (node.left != null) { reverse.add(node.left); } if (node.right != null) { reverse.add(node.right); } } // 当前是从右往左打印的,那就按从右往左入栈 else { if (node.right != null) { reverse.add(node.right); } if (node.left != null) { reverse.add(node.left); } } if (current.size() == 0) { flag = 1 - flag; List<BinaryTreeNode> tmp = current; current = reverse; reverse = tmp; System.out.println(); } } } }
Copyright© 2013-2020
All Rights Reserved 京ICP备2023019179号-8