给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1
/ \
2 3
\ \
5 4
解题思路:
首先判断是否有root,没有则输出[]
然后设置两个列表queue,res前者存放需要搜索的节点,后者为值
接着开始循环queue,再量取queue的值,有几就循环几次,然后取出值。接着判断是否有左右节点,有的话就加入queue中。
当循环完了以后,将最后取出的一个cur以cur.val的值加入到res中。因为是先加左节点再加右节点,所以取出的值依次是从左到右,那么有右节点最后一次取出就是右节点,没有的话最后一次取出就是左节点。
最后输出res。
Python代码:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def rightSideView(self, root: TreeNode) -> List[int]:
res = []
if not root:
return res
queue = collections.deque()
queue.append(root)
while queue:
for _ in range(len(queue)):
node = queue.popleft()
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
res.append(node.val)
return res
原文链接:https://blog.csdn.net/weixin_42295205/article/details/113250584