LeetCode build tree questions

二叉树的建树和序列化问题有很多,
中序加后序建树
https://leetcode.com/problems/construct-binary-tree-from-inorder-and-postorder-traversal/
前序建搜索树
https://leetcode.com/problems/construct-binary-search-tree-from-preorder-traversal/
前序加后序建树
https://leetcode.com/problems/construct-binary-tree-from-preorder-and-postorder-traversal/
前序加中序建树
https://leetcode.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/

基本的思路其实还是用递归法来解决这些问题,中序加后序的,我们从后向前看,把中序和后序每次分成两部分,继续使用递归去解决找到子树。

    def buildTree(self, inorder: List[int], postorder: List[int]) -> TreeNode:
        if not inorder:
            return None
        
        posDict = collections.defaultdict(int)
        for i, val in enumerate(inorder):
            posDict[val] = i

        return Solution.buildTreeHelp(inorder, 0, len(inorder) -1, postorder, posDict)
        
    @staticmethod
    def buildTreeHelp(inorder: List[int], si: int, ei: int, postorder: List[int], posDict: dict) -> TreeNode:
        if si > ei:
            return None

        root = TreeNode(postorder[-1])
        root.val = postorder[-1]
        ipos = posDict[root.val]
        root.left = Solution.buildTreeHelp(inorder, si, posDict[root.val] -1, postorder[0: ipos-si], posDict)
        root.right = Solution.buildTreeHelp(inorder, ipos +1, ei, postorder[ipos-si: -1], posDict)

        return root    

前序建搜索树,可以通过不断的插入新节点来重建

    def bstFromPreorder(self, preorder: List[int]) -> TreeNode:
        if preorder == None or len(preorder) == 0: return None

        root = TreeNode(preorder[0])

        def insertNode(node, val):
            if val < node.val:
                if node.left == None:
                    node.left = TreeNode(val)
                else:
                    insertNode(node.left, val)
            else:
                if node.right == None:
                    node.right = TreeNode(val)
                else:
                    insertNode(node.right, val)        
        
        for i in range(1, len(preorder)):
            insertNode(root, preorder[i])
        return root              

前序加后序,这个稍微麻烦一点,但是根本的原理还是,利用递归,不断的把两个序列拆成两部分来建立左右子树。

    def constructFromPrePost(self, pre: List[int], post: List[int]) -> TreeNode:
        if len(pre) == 0 or len(post) == 0: return None

        if pre[0] == post[-1]:
            currentNode = TreeNode(pre.pop(0))
            post.pop()

        if len(pre) == 0 or len(post) == 0:
            return currentNode

        try:
            index = post.index(pre[0])
        except ValueError:
            return currentNode

        currentNode.left = self.constructFromPrePost(pre, post[0:index+1])
        currentNode.right = self.constructFromPrePost(pre, post[index + 1:])

        return currentNode

前序加中序建树,其实和后续加中序是非常类似的,稍微换个写法,代码更加简单:

    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        if len(preorder) == 0 or len(inorder) == 0: 
            return None

        currentNode = TreeNode(preorder.pop(0))
        index = inorder.index(currentNode.val)

        currentNode.left = self.buildTree(preorder, inorder[0:index])
        currentNode.right = self.buildTree(preorder, inorder[index+1:])
        return currentNode        
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值