输出二叉树的右视图

输出二叉树的右视图-牛客

思路

1、建树
递归构建左右子树

1）递归函数入参和返回值：入参：前序数组、中序数组；返回值：root
2）终止条件：前序数组长度为0或中序数组长度为0
3）单层逻辑：先利用前序的首位找到根节点，再根据根节点，遍历找到中序数组中根节点的下标，依据下标划分左子树的前序、中序结果，右子树的前序、中序结果，最后递归调用构建二叉树。

2、层序遍历获取每一层最右节点的值
把每一层结点放入双头队列que中，循环条件: 队列长度不为0
当队列长度==0，表示已到本层最右结点，把最右结点的值放入结果数组中。

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 求二叉树的右视图
# @param xianxu int整型一维数组 先序遍历
# @param zhongxu int整型一维数组 中序遍历
# @return int整型一维数组
#
class Solution:
    def solve(self , xianxu: List[int], zhongxu: List[int]) -> List[int]:
        # 二叉树递归+层序遍历+双头队列
        #空处理
        if len(xianxu)==0 or len(zhongxu)==0:
            return []
        #1、建树
        root=self.buildTree(xianxu, zhongxu)
        #2、打印右视图：每层最右边结点，层序遍历
        return self.rightView(root)
        
    def buildTree(self, xianxu, zhongxu):
        n=len(xianxu)
        m=len(zhongxu)
        if n==0 or m==0:
            return None
        root=TreeNode(xianxu[0])
        for i in range(len(zhongxu)):
            if zhongxu[i]==xianxu[0]:
                leftpre=xianxu[1:i+1]
                leftvin=zhongxu[:i]
                rightpre=xianxu[i+1:]
                rightvin=zhongxu[i+1:]
                #构建左、右子树
                root.left=self.buildTree(leftpre, leftvin)
                root.right=self.buildTree(rightpre, rightvin)
                break
        return root
    
    def rightView(self,root):
        #层序遍历+双头队列
        res=[]
        from collections import deque
        que=deque([root])
        while que:
            #本层的节点数
            size=len(que)
            while size:
                size-=1
                cur=que.popleft()
                if cur.left:
                    que.append(cur.left)
                if cur.right:
                    que.append(cur.right)
                #最右节点
                if size==0:
                    res.append(cur.val)
        return res