네 번째 문제: 두 갈래 나무 재건

2966 단어 검지offer
두 갈래 나무의 앞 순서와 중간 순서의 결과를 입력하십시오. 이 두 갈래 나무를 다시 만드십시오.입력한 앞 순서와 중간 순서의 결과에 중복된 숫자가 없다고 가정하십시오.예를 들어 앞 순서 반복 시퀀스 {1,2,4,7,3,5,6,8}와 중간 순서 반복 시퀀스 {4,7,2,1,5,3,8,6}를 입력하면 두 갈래 트리를 재건하고 되돌려줍니다.
# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def reConstructBinaryTree(self, pre, tin):
        if not pre or not tin:
            return None
        root = TreeNode(pre.pop(0))  # , pre 
        index = tin.index(root.val)  # 
        root.left = self.reConstructBinaryTree(pre, tin[:index])  # 
        root.right = self.reConstructBinaryTree(pre, tin[index + 1:]) # 
        return root

좋은 웹페이지 즐겨찾기