leetcode 1008 Construct Binary Search Tree from Preorder Traversal
leetcode 1008 Construct Binary Search Tree from Preorder Traversal
1.题目描述
返回与给定先序遍历 preorder
相匹配的二叉搜索树(binary search tree
)的根结点。
(回想一下,二叉搜索树是二叉树的一种,其每个节点都满足以下规则,对于 node.left
的任何后代,值总 < node.val
,而 node.right
的任何后代,值总 > node.val
。此外,先序遍历首先显示节点的值,然后遍历 node.left
,接着遍历 node.right
。)
示例:
输入:[8,5,1,7,10,12]
输出:[8,5,10,1,7,null,12]
提示:
1 <= preorder.length <= 100
先序 preorder 中的值是不同的。
2.解题思路
根据二叉搜索树的规则,第一个数字一定是父节点,遍历之后的数字,当出现第一个大于父节点的数字后,这个数字以后的全部在右子树,之前的全部在左子树。递归实现即可。
3.Python代码
class Solution(object):
def bstFromPreorder(self, preorder):
"""
:type preorder: List[int]
:rtype: TreeNode
"""
if not preorder:
return None
root=TreeNode(preorder[0])
seg=len(preorder)
for i in range(1,len(preorder)):
if preorder[i]>root.val:
seg=i
break
root.left=self.bstFromPreorder(preorder[1:seg])
root.right=self.bstFromPreorder(preorder[seg:])
return root