二叉树的锯齿形层次遍历
给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回锯齿形层次遍历如下:
[
[3],
[20,9],
[15,7]
]
解题思路:锯齿状遍历核心思想就是根据当前层数的奇偶来判断遍历后数字的顺序。因此可以采取使用两个栈的方式,根据层数判断使用哪个栈,并且根据层数判断压栈的顺序,然后将每一层的值直接出栈放入List中即可。
/*** Definition for a binary tree node.* public class TreeNode {* int val;* TreeNode left;* TreeNode right;* TreeNode(int x) { val = x; }* }*/class Solution {public List<List<Integer>> zigzagLevelOrder(TreeNode root) {// 解题思路,双栈存储不同层级的值,存储时如果需要左到右,则右到左入栈,如果需要右到左,则左到右入栈// 1.创建返回结果List<List<Integer>> res = new ArrayList<>();if (root == null) {return res;}// 2.创建第一个栈,用于存储奇数层的值Stack<TreeNode> stack1 = new Stack<>();// 3.创建第二个栈,用于存储偶数层的值Stack<TreeNode> stack2 = new Stack<>();// 4.将第一个奇数层值放入栈中,即把根节点放入栈中stack1.push(root);// 5.判断当前两个栈是否均为空,均为空说明已经遍历完整个树,因为遍历过程中,stack1为空,则stack2不为空while(!stack1.isEmpty() || !stack2.isEmpty()){// 创建一个存储当前层数据的listList<Integer> list = new ArrayList<>();// 获得当前节点TreeNode cur = null;// 若第一个栈不为空,循环遍历第一个栈,把元素输出为一个list存储if(!stack1.isEmpty()){while(!stack1.isEmpty()){cur = stack1.pop();list.add(cur.val);if(cur.left!=null){stack2.push(cur.left);}if(cur.right!=null){stack2.push(cur.right);}}}else{while(!stack2.isEmpty()){cur = stack2.pop();list.add(cur.val);if(cur.right != null){stack1.push(cur.right);}if(cur.left != null){stack1.push(cur.left);}}}res.add(list);}return res;}}
