思路:右侧的节点就是每一层的最后一个元素,所以采用BFS算法,把每一次的最后一个节点加入到结果列表中即可。需要初始化一个队列,这个而且放入队列的顺序是先左在右.那么最后一个元素必定是右侧节点值。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def rightSideView(self, root: TreeNode) -> List[int]:
if root == None:
return []
q = collections.deque() #初始话一个对垒
res = [] #结果列表
q.append(root) #将跟节点加入对列中
while q:
length = len(q) #列表长度
temp = 0
for i in range(0,length):
item = q.popleft() #取出列表中的第一个元素
temp = item.val
if item.left: #先添加左子树节点,在添加右子树节点
q.append(item.left)
if item.right:
q.append(item.right)
res.append(temp)
return res
总结:这个题目是要返回最有边的节点,所以如果用层次遍历+BFS的话,取出每次最后一个元素就可以了。