上次的广度搜索优先的题目做完之后,看了下别人的代码,发现写的非常不错。这次就自己变化了一点用在了新题上,代码如下:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def rightSideView(self, root):
"""
:type root: TreeNode
:rtype: List[int]
"""
if not root:
return []
a = []
queue = [root]
while queue:
a.append([q.val for q in queue][-1])
queue = [q for node in queue for q in (node.left, node.right) if q]
return a