leetcode 889. 전차와 후차에 따라 두 갈래 나무를 두루 구성하다

4558 단어 leetcode
889. 전순과 후순에 따라 2차원 트리를 구성하여 주어진 전순과 후순이 일치하는 2차원 트리를 되돌려줍니다.
pre와post가 두루 다니는 값은 서로 다른 정수입니다.
예:
입력: pre = [1,2,4,5,3,6,7],post = [4,5,2,6,7,3,1] 출력: [1,2,3,4,5,6,7]
팁:
1 <= pre.length == post.length<=30 pre[]와post[]는 모두 1, 2,...,pre.length의 배열은 입력마다 적어도 하나의 답이 있어야 한다.만약 여러 개의 답이 있다면, 그 중 하나로 돌아갈 수 있다.
이 문제와 이전의 사용 중 전;중후는 유사하지만 이곳의 사용 전후에 유일하게 왼쪽 나무와 오른쪽 나무를 확정할 수 없으며, 우리는 왼쪽 나무의 뿌리 노드pre[1]를 처리할 때 귀속 중의pre의 길이를 고려해야 한다.
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def constructFromPrePost(self, pre: List[int], post: List[int]) -> TreeNode:
        if not pre and not post: return
        if len(pre) == 1: return TreeNode(pre[0])

        index = post.index(pre[1])+1# , 
        root = TreeNode(pre[0])
        root.left = self.constructFromPrePost(pre[1:index+1], post[:index])
        root.right = self.constructFromPrePost(pre[index+1:], post[index:-1])
        return root
        

좋은 웹페이지 즐겨찾기