검지 Offer-7 두 갈래 나무 재건

5603 단어

제목:


두 갈래 나무의 앞 순서와 중간 순서의 결과를 입력하십시오. 이 두 갈래 나무를 다시 만드십시오.입력한 앞 순서와 중간 순서의 결과에 중복된 숫자가 없다고 가정하십시오.예를 들어 앞 순서 반복 시퀀스 {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:
    #  TreeNode 
    def reConstructBinaryTree(self, pre, tin):
        # write code here
        if not pre or not tin or len(pre) != len(tin): return
        return self.reconstruct(pre, tin, 0, len(pre)-1, 0, len(tin)-1)
    def reconstruct(self, pre, tin, preStart, preEnd, tinStart, tinEnd):
        rootValue = pre[preStart]
        rootIndex = tin.index(rootValue)
        root = TreeNode(rootValue)
        if preStart == preEnd:
            return root
        if rootIndex > tinStart:
            root.left = self.reconstruct(pre, tin, preStart + 1, rootIndex - tinStart + preStart, tinStart, rootIndex - 1)
        if tinEnd > rootIndex:
            root.right = self.reconstruct(pre, tin, rootIndex - tinStart + preStart + 1, preEnd, rootIndex + 1, tinEnd)
        return root

좋은 웹페이지 즐겨찾기