package Stack;
import java.util.ArrayList;
import java.util.List;
import java.util.Stack;
public class ZigzagLevelOrder_103 {
/*
* 给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
*
* 例如: 给定二叉树 [3,9,20,null,null,15,7],
*
* 3 / \ 9 20 / \ 15 7
*
* 返回锯齿形层次遍历如下:
*
* [ [3], [20,9], [15,7] ]
*
* 来源:力扣(LeetCode)
* 链接:https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
* 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
*/
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
// 思路:
// 1、用两个栈交替存储(因为用一个栈判断一行结束比较麻烦)
// 2、如果子节点不为空则入栈,偶数行从右至左入栈,奇数行从左至右入栈
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
Stack<TreeNode> stack1 = new Stack<>();
Stack<TreeNode> stack2 = new Stack<>();
List<List<Integer>> res = new ArrayList<>();
// 判断为空的情况
if(root==null){
return res;
}
int flag = 1;
stack1.push(root);
while (!stack1.isEmpty() || !stack2.isEmpty()) {
List<Integer> list = new ArrayList<>();
if (flag == 1) {
while (!stack1.isEmpty()) {
root = stack1.pop();
list.add(root.val);
if (root.left != null)
stack2.push(root.left);
if (root.right != null)
stack2.push(root.right);
}
res.add(list);
flag = 0;
}
if (flag == 0) {
while (!stack2.isEmpty()) {
root = stack2.pop();
list.add(root.val);
if (root.right != null)
stack1.push(root.right);
if (root.left != null)
stack1.push(root.left);
}
res.add(list);
flag = 1;
}
}
return res;
}
}
103. 二叉树的锯齿形层次遍历
最新推荐文章于 2022-11-16 11:58:36 发布