主体思想
通过递归返回节点的方式,注意当这个节点值在范围内的时候就可以继续返回,只需要注意下它左右节点的赋值即可。如果不在范围内的时候,这个时候二叉树的特性就展露出来了
不在范围内的可能只有两种:
1.在这个区间的左边
因为二叉搜索树的特性,在左边代表着这个节点和它左边的所有节点全都小于,这时只需把右边校验好的节点返回回去就行
2.在这个区间的右边
因为二叉搜索树的特性,在右边代表着这个节点和它右边的所有节点全都大于,这时只需把左边校验好的节点返回回去就行
代码实现
func trimBST(root *TreeNode, low int, high int) *TreeNode {
if root==nil{
return nil
}
if root.Val>=low && root.Val<=high{
root.Left=trimBST(root.Left,low,high)
root.Right=trimBST(root.Right,low,high)
return root
}else if root.Val<low{
return trimBST(root.Right,low,high)
}else{
return trimBST(root.Left,low,high)
}
}