【leetcode】103.（Medium）Binary Tree Zigzag Level Order Traversal

最新推荐文章于 2021-02-05 11:10:39 发布

原创最新推荐文章于 2021-02-05 11:10:39 发布 · 137 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

257 篇文章

订阅专栏

本文介绍了一种使用两个栈实现的二叉树锯齿形层序遍历算法，通过交替使用两个栈来达到逆序输出奇数层节点的效果。此算法的时间复杂度为O(n)，空间复杂度为O(n)。

解题思路：
维护两个栈，轮番记录每一层的结点

时间复杂度：O(n)
空间复杂度：O(n)

提交代码：

class Solution {
	    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
	        List<List<Integer>> res=new ArrayList<List<Integer>>();
	        if(root==null)	return res;
	        
	        Stack<TreeNode> s1=new Stack<>();
	        Stack<TreeNode> s2=new Stack<>();
	        s1.push(root);
	        
	        while(s1.size()!=0||s2.size()!=0) {
	        	List<Integer> level1=new ArrayList<Integer>();
	        	while(s1.size()!=0) {
	        		TreeNode tmp=s1.pop();
	        		level1.add(tmp.val);
	        		if(tmp.left!=null)
	        			s2.push(tmp.left);
	        		if(tmp.right!=null)
	        			s2.push(tmp.right);
	        	}
	        	if(level1.size()!=0)
	        		res.add(level1);
	        	
	        	List<Integer> level2=new ArrayList<>();
	        	while(s2.size()!=0) {
	        		TreeNode tmp=s2.pop();
	        		level2.add(tmp.val);
	        		if(tmp.right!=null)
	        			s1.add(tmp.right);
	        		if(tmp.left!=null)
	        			s1.add(tmp.left);
	        	}
	        	if(level2.size()!=0)
	        		res.add(level2);
	        }
	        return res;
	    }
	}