669. 修剪二叉搜索树

主体思想

  通过递归返回节点的方式,注意当这个节点值在范围内的时候就可以继续返回,只需要注意下它左右节点的赋值即可。如果不在范围内的时候,这个时候二叉树的特性就展露出来了
不在范围内的可能只有两种:
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)
    }
}

你可能感兴趣的:(算法,算法)