您的位置:首页 > 其它

【LeetCode OJ】Binary Tree Zigzag Level Order Traversal

2014-05-27 04:36 387 查看
Problem Link:

https://oj.leetcode.com/problems/binary-tree-zigzag-level-order-traversal/

Just BFS from the root and for each level insert a list of values into the result.

# Definition for a  binary tree node
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
# @param root, a tree node
# @return a list of lists of integers
def zigzagLevelOrder(self, root):
"""
BFS from the root.
For each level, insert a list of values into the result
"""
res = []
if not root:
return res
q = [root]
reverse = False
while q:
new_q = []
if reverse:
res.append([n.val for n in q][::-1])
else:
res.append([n.val for n in q])
reverse = not reverse
for node in q:
if node.left:
new_q.append(node.left)
if node.right:
new_q.append(node.right)
q = new_q
return res
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: