码迷,mamicode.com
首页 > 其他好文 > 详细

【LeetCode每天一题】Binary Tree Zigzag Level Order Traversal(二叉树的之字形遍历)

时间:2019-05-13 12:40:32      阅读:130      评论:0      收藏:0      [点我收藏+]

标签:treenode   层遍历   pen   思路   obj   for   添加   应该   The   

Given a binary tree, return the zigzag level order traversal of its nodes‘ values. (ie, from left to right, then right to left for the next level and alternate between).

For example:
Given binary tree [3,9,20,null,null,15,7],

    3
   /   9  20
    /     15   7

 

return its zigzag level order traversal as:

[
  [3],
  [20,9],
  [15,7]
]

思路

  对于二叉树的之字形遍历,我们可以使用像之前对二叉树进行层次遍历时一样,只不过这里对每一层节点在添加子节点时的顺序有要求,当为基数层时,我们在添加节点时选择先添加左边的子节点,在添加右边的子节点。 当为偶数层时,我们应该先添加右边的子节点,在添加左边的子节点。然后遍历完毕之后在将该层节点的遍历结果添加到最终列表中。
解决代码


 1 # Definition for a binary tree node.
 2 # class TreeNode(object):
 3 #     def __init__(self, x):
 4 #         self.val = x
 5 #         self.left = None
 6 #         self.right = None
 7 
 8 class Solution(object):
 9     def zigzagLevelOrder(self, root):
10         """
11         :type root: TreeNode
12         :rtype: List[List[int]]
13         """
14         if not root:
15             return []
16         
17         res, index = [], 0  # 结果列表 和  当前层数变量
18         stack = [root]      # 使用辅助栈
19         while stack:
20             tem_stack, count = [], len(stack)     # 申请一个空列表来存储当前层的子节点, count表示当前层中的节点个数
21             level, index = [], index + 1        # level 表示存储当前层的节点值的遍历顺序, 
22             for _ in range(count):              # 当前层有多少个节点就遍历多少次
23                 tem = stack.pop()              # 弹出顶部元素
24                 if index % 2 == 1:              # 判断当前层,然后选择不同的子节点添加规则。
25                     if tem.left:                # 奇数层先添加左节点,在添加右节点
26                         tem_stack.append(tem.left)
27                     if tem.right:
28                         tem_stack.append(tem.right)
29                 else:                           # 偶数层先添加右节点在添加左节点。
30                     if tem.right:
31                         tem_stack.append(tem.right)
32                     if tem.left:
33                         tem_stack.append(tem.left)
34                         
35                 level.append(tem.val)          #  添加当前节点的值
36             stack =  tem_stack                 # 当前层遍历结束之后,将tem_stack赋值给stack.(tem_stack表示下一层子节点的排列顺序)
37             res.append(level)                  # 当前层的遍历结果添加进去
38         return res
39                    

【LeetCode每天一题】Binary Tree Zigzag Level Order Traversal(二叉树的之字形遍历)

标签:treenode   层遍历   pen   思路   obj   for   添加   应该   The   

原文地址:https://www.cnblogs.com/GoodRnne/p/10855890.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!