# 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 classSolution: defrightSideView(self, root: Optional[TreeNode]) -> List[int]: if root isNone: return [] ans = [] q = deque([root]) while q: # 队列中的最后一个元素即为当前层最右侧的节点 ans.append(q[-1].val) # 遍历并处理当前层的所有节点 for _ inrange(len(q)): node = q.popleft() if node.left: q.append(node.left) if node.right: q.append(node.right) return ans
classTreeNode: def__init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right
defbuild_tree(arr): ifnot arr or arr[0] isNone: returnNone root = TreeNode(arr[0]) queue = [root] i = 1 while queue and i < len(arr): node = queue.pop(0) if arr[i] isnotNone: node.left = TreeNode(arr[i]) queue.append(node.left) i += 1 if i < len(arr) and arr[i] isnotNone: node.right = TreeNode(arr[i]) queue.append(node.right) i += 1 return root
deftree_to_list(root): ifnot root: return [] result = [] queue = [root] while queue: node = queue.pop(0) if node: result.append(node.val) queue.append(node.left) queue.append(node.right) else: result.append(None) while result and result[-1] isNone: result.pop() return result
classSolution: defrightSideView(self, root: Optional[TreeNode]) -> List[int]: if root isNone: return [] ans = [] q = deque([root]) while q: # 队列中的最后一个元素即为当前层最右侧的节点 ans.append(q[-1].val) # 遍历并处理当前层的所有节点 for _ inrange(len(q)): node = q.popleft() if node.left: q.append(node.left) if node.right: q.append(node.right) return ans
defmain(): arr = sys.stdin.read().strip().split() tree_arr = [Noneif v == 'None'elseint(v) for v in arr] root = build_tree(tree_arr)
sol = Solution() result = sol.rightSideView(root) print(result)