leetcode699-修剪二叉搜索树
leetcode 699
思路
本题可能存在左节点值 < low但是左节点的右子树仍然>low的情况,还可能存在右节点值 > high 但是右节点的左子树仍然 < high的情况,所以当我们判断出当前root不在[low,high]区间的时候,还需要对子树进行递归,以确保满足条件的子树不被修剪
实现
var trimBST = function (root, low, high) {
if (!root) return null;
if (root.val < low) {
// 右子树是大于当前节点的,所以可能右子树会满足,需要进行递归
return trimBST(root.right, low, high)
} else if (root.val > high) {
// 左子树是小于当前节点的,所以可能左子树满足范围,需要进行递归
return trimBST(root.left, low, high)
}
root.left = trimBST(root.left, low, high)
root.right = trimBST(root.right, low, high)
return root;
};