leetcode 105 Construct Binary Tree from Preorder and Inorder Traversal

Java:

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
      return build(preorder, 0, inorder, 0, inorder.length-1);  
    }
    
    public TreeNode build(int[] preorder, int pre_str, int[] inorder, int in_str, int in_end){
        if(pre_str > preorder.length-1 || in_str > in_end) return null;
        int index = 0;
        TreeNode root = new TreeNode(preorder[pre_str]);
        for(int i=in_str; i<=in_end; i++){
            if(inorder[i] == root.val){
                index = i;
                break;
            }
        }
        root.left = build(preorder, pre_str+1, inorder, in_str, index-1);
        root.right = build(preorder, pre_str+index-in_str+1, inorder, index+1, in_end);
        return root;
    }
}

Python:

class Solution:
    def buildTree(self, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: List[int]
        :rtype: TreeNode
        """
        if inorder:
            beg = preorder.pop(0)
            #.index是否包含字符
            ind = inorder.index(beg)
            root = TreeNode(inorder[ind])
            root.left = self.buildTree(preorder, inorder[0:ind])
            root.right = self.buildTree(preorder, inorder[ind+1:])
            return root


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值