[LeetCode Python3] 103. Binary Tree Zigzag Level Order Traversal + 两种解法

SincX

于 2020-10-20 18:58:34 发布

阅读量165

点赞数

分类专栏： LeetCode每日一题

本文链接：https://blog.csdn.net/xvrixingkong/article/details/109186893

版权

LeetCode每日一题专栏收录该内容

172 篇文章 0 订阅

订阅专栏

103. Binary Tree Zigzag Level Order Traversal

S1:

# V1 
class Solution:
    def zigzagLevelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:
            return None
        res, restmp = [], []
        stack0, stack1 = [root], []
        flag = True
        while stack0:
            tmp = stack0.pop()
            restmp.append(tmp.val)
            if flag:
                if tmp.left:
                    stack1.append(tmp.left)
                if tmp.right:
                    stack1.append(tmp.right)
            else:
                if tmp.right:
                    stack1.append(tmp.right)
                if tmp.left:
                    stack1.append(tmp.left)
            if not stack0:
                stack0 = stack1[:]
                stack1.clear()
                flag = not flag
                res.append(restmp[:])
                restmp.clear()
        return res

S2:

class Solution:
    def zigzagLevelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:
            return None
        res, restmp = [], []
        stack0, stack1 = [root], []
        flag = True
        while stack0:
            for tmp in stack0:
                restmp.append(tmp.val)
                if tmp.left:
                    stack1.append(tmp.left)
                if tmp.right:
                    stack1.append(tmp.right)
            stack0 = stack1[:]
            stack1.clear()
            if not flag:
                restmp.reverse()
            flag = not flag
            res.append(restmp[:])
            restmp.clear()
        return res