📜  打印二叉树的右下角视图(1)

📅  最后修改于: 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为二叉树节点数。