LeetCode 199. 二叉树的右视图（Binary Tree Right Side View）

• 原题
• 思路
• 代码

原题

给定一棵二叉树，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。

示例:

输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:

1            <---
/   \
2     3         <---
\     \
5     4       <---

思路

二叉树层序遍历，输出每层最后一个元素。

代码

# 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 []
res = []
p1 = [root]
while p1:
res.append(p1[-1].val)
p = []
for node in p1:
if node.left: p.append(node.left)
if node.right: p.append(node.right)
p1 = p
return res