题解 | #输出二叉树的右视图#

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
# 求二叉树的右视图
# @param xianxu int整型一维数组 先序遍历
# @param zhongxu int整型一维数组 中序遍历
# @return int整型一维数组
#
class Solution:
    def solve(self , xianxu , zhongxu ):
        # write code here
        result = []
        def dfs(p, i, level):
            if not p:
                return
            if level >= len(result):
                result.append(p[0])
            else:
                result[level] = p[0]

            tmp = i.index(p[0])  
            dfs(p[1:tmp+1], i[0:tmp] , level+1)
            dfs(p[tmp+1:], i[tmp+1:], level+1)

        dfs(xianxu, zhongxu, 0) 
        return result