不简单的递归
如果当前结点不符合要求,有两种情况
1.小于low,所以root和左子树都不要了,要的是修改后的右子树,递归修改右子树然后返回。
2.大于high
如果正常,就递归修改自己的左边和右边,然后返回root
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int low, int high) {
if(root == nullptr) return root;
//判断当前结点是否符合要求,如果小于low,那么就直接返回右子树(当然是修剪过后的右子树)
//如果大于high,就返回修剪过后的左子树,必须修剪,因为左子树也不一定全符合要求,但是root和右子树一定不符合要求
if(root->val < low) return trimBST(root->right,low,high);
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;
}
};