(1)使用一个双端队列,每次先进左节点,再进右节点,然后将队列最左侧的元素弹出(二叉树最右边的元素)。
(2)要始终维持一个最大深度,以便于最后返回字典里面的值。
(3)创建的字典,队列都是(节点/节点的值, 深度)这样一个对应关系,保存节点或者节点的值的同时要保存深度。
from collections import deque
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):
max_depth = -1
right_value = dict()
queue = deque([(root , 0)])
while queue:
node, depth = queue.popleft()
if node is not None:
max_depth = max(depth, max_depth)
right_value[depth] = node.val
queue.append((node.left, depth+1))
queue.append((node.right, depth+1))
return [right_value[depth] for depth in range(max_depth+1)]
node = TreeNode(1)
node.left = TreeNode(4)
node.right = TreeNode(10)
node.right.left= TreeNode(2)
a = Solution()
b = a.rightSideView(node)
print(b)