Given a binary tree, imagine yourself standing on the right side of it, return the values of the nodes you can see ordered from top to bottom.
Example:
Input: [1,2,3,null,5,null,4] Output: [1, 3, 4] Explanation: 1 <--- / \ 2 3 <--- \ \ 5 4 <---
这道题很简单,记录层序遍历后最右面的点。
层序遍历用BFS算法就很简单,然后保存每一层最后一个值就可以。用tempval保存每一层的val,tempnode保存每一层的node,final保存每一层最右面的一个val。
class Solution:
def rightSideView(self, root: TreeNode) -> List[int]:
if root is None: return root
bfs=[root]
final=[]
while bfs:
tempval=[node.val for node in bfs]
final.append(tempval[-1])
tempnode=[]
for node in bfs:
if node.left:
tempnode.append(node.left)
if node.right:
tempnode.append(node.right)
bfs = tempnode
return final