剑指 Offer 07. 重建二叉树

时间复杂度O(nlogn),可以有重复元素的解法:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
    	if len(preorder)==0 or len(inorder)==0:return
    	root=TreeNode(preorder[0])
    	idx=inorder.index(preorder[0])#list.index()返回第一个符合要求的值的index
    	root.left=self.buildTree(preorder[1:idx+1],inorder[0:idx])
    	root.right=self.buildTree(preorder[idx+1:],inorder[idx+1:])
    	return root

时间复杂度为O(n)的解法,只适用于没有重复元素:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
    	dic={}
    	for index,val in enumerate(inorder):
    		dic[val]=index 
    	def helper(preorder_left,preorder_right,inorder_left,inorder_right):
    		if preorder_left>preorder_right:return
    		root=TreeNode(preorder[preorder_left])
    		root_inorder=dic[preorder[preorder_left]]
    		left_size=root_inorder-inorder_left
    		root.left=helper(preorder_left+1,preorder_left+left_size,inorder_left,root_inorder-1)
			root.right=helper(preorder_left+left_size+1,preorder_right,root_inorder+1,inorder_right)
			return root
		return helper(0,len(preorder)-1,0,len(inorder)-1)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值