[검지 offer] 10 - 두 갈래 나무 재건

3129 단어
제목:
두 갈래 나무의 앞 순서와 중간 순서의 결과를 입력하십시오. 이 두 갈래 나무를 다시 만드십시오.
입력한 앞 순서와 중간 순서의 결과에 중복된 숫자가 없다고 가정하십시오.
예를 들어 앞 순서 반복 시퀀스 {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:
            return None
        if len(pre) <= 0 or len(tin) <= 0:
            return None
        root = TreeNode(pre[0])  #  , treenode 
        pos = tin.index(pre[0])  #  

        tinLeft = tin[:pos]  #  
        tinRight = tin[pos+1:]

        preLeft = pre[1:pos+1]  #  
        preRight = pre[pos+1:]

        #  , , , 
        root.left = self.reConstructBinaryTree(preLeft, tinLeft)
        root.right = self.reConstructBinaryTree(preRight, tinRight)
        return root

쓸 때 겪는 문제:
1. 특수한 상황에 대한 고려: 전순/중순이 없는 경우.앞/가운데 길이 0인 경우
2. pre[0]는 뿌리 노드지만 두 갈래 나무 TreeNode에 넣어야 합니다!TreeNode의 left는 왼쪽 트리, right는 오른쪽 트리
3. 루트는 이제 TreeNode 클래스인 만큼 left와 right가 있습니다.

좋은 웹페이지 즐겨찾기