- 修剪二叉搜索树
- 思路:
递归:参数为要处理的当前节点以及区间 [low, high],返回值为当前节点经过处理后返回的节点;
当前节点看作根节点;目的是让每个非空结点值都在区间内;
终止情况:找到空结点,或该节点值在区间内
- root为空直接返回NULL;
- root不为空:
要么左边出界:root->val很小,则root的左子树值更小就不用看了,但root的右子树中仍可能找到符合要求的结点,因此将root->right看作新的根节点继续去找,并返回给上一层连接
要么右边出界:同理,将root->left看作新的根节点继续去找,并返回给上一层连接
要么不出界,root->val就在区间内,则分别寻找root的左孩子和右孩子,return root;
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int low, int high) {
if (!root) return NULL;
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;
}
};
- 对于递归单层逻辑写的过程中,不要往后想太多,很多时候,递归就是一层一层的调用自己就得到答案了,不需要自己加很多种情况的判断;