📅  最后修改于: 2023-12-03 15:25:45.134000             🧑  作者: Mango
给定一棵二叉树,打印其右下角的节点值。
例如,下图中箭头所指的节点值为7,因此右下角节点值为7。
从右边开始对二叉树进行遍历,每次记录当前层的最后一个节点值即可。具体实现可以采用广度优先搜索(BFS)进行层次遍历。
class Solution(object):
def rightSideView(self, root):
"""
:type root: TreeNode
:rtype: List[int]
"""
if not root:
return []
queue = [root]
res = []
while queue:
size = len(queue)
for i in range(size):
node = queue.pop(0)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
if i == size - 1:
res.append(node.val)
return res
class Solution {
public List<Integer> rightSideView(TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) {
return res;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
int size = queue.size();
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
if (i == size - 1) {
res.add(node.val);
}
}
}
return res;
}
}
上述代码片段分别为Python和Java语言的实现。其中,queue
为队列,初始时将根节点入列;res
为结果列表,用于存储每层的最后一个节点值;size
用于记录当前层的节点个数。在单次循环中,从队列中出列当前层的节点并将其左右子节点入列,同时记录当前层的最后一个节点值,最后将其添加到res
中并返回。理论上,每个节点仅会遍历一次,因此时间复杂度为O(n),其中n为二叉树节点数。