题目描述:
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历结果。如果是则返回 true,否则返回 false。假设输入的数组的任意两个数字都互不相同。
参考以下这颗二叉搜索树:
示例 1:
输入: [1,6,3,2,5]
输出: false
示例 2:
输入: [1,3,2,6,5]
输出: true
解题思路:
1.后序遍历最后一个数字即为根节点
2.二叉搜索树,左子树全部小于根节点,右子树全部大于根节点
3.找到第一个大于root的节点,如果其后面节点均大于root,符合条件,递归判断左子树和右子树,否则返回false
class Solution:
def verifyPostorder(self, postorder: List[int]) -> bool:
#print(postorder)
if len(postorder)==1 or len(postorder)==0:
return True
root=postorder[-1]
left,right=[],[]
for i in range(len(postorder)-1):
node=postorder[i]
if node<root:
left.append(node)
elif node==root:
return False
else:
#print(i,postorder[i:-1])
if min(postorder[i:-1])<=root:
return False
else:
right=postorder[i:-1]
break
#print(left,right)
return self.verifyPostorder(left) and self.verifyPostorder(right)
class Solution:
def verifyPostorder(self, postorder: List[int]) -> bool:
def helper(nums):
if len(nums) <= 1: return True
root = nums[-1]
for i in range(len(nums)):
if nums[i] > root: break
for j in range(i, len(nums)-1):
if nums[j] < root: return False
return helper(nums[:i]) and helper(nums[i:-1])
if not postorder: return True
return helper(postorder)