LeetCode669修剪二叉搜索树
题目链接
https://leetcode-cn.com/problems/trim-a-binary-search-tree/
题解
- 递归解法
- 我看了题解写出来的,我分析题目的时候以为还要交换左右子树什么的……,其实不用
- 思路见代码注释
// Problem: LeetCode 669
// URL: https://leetcode-cn.com/problems/trim-a-binary-search-tree/
// Tags: Tree BST Recursion
// Difficulty: Easy
struct TreeNode{
int val;
TreeNode* left;
TreeNode* right;
};
class Solution{
public:
TreeNode* trimBST(TreeNode* root, int L, int R) {
// 处理空链表
if (root == nullptr) return nullptr;
// 如果根结点小于L,则根结点以及左子树会被剪掉,整个树的修剪结果是修剪后的右子树
if (root->val < L) return trimBST(root->right, L, R);
// 如果根结点小于L,则根结点以及右子树会被剪掉,整个树的修剪结果是修剪后的左子树
if (root->val > R) return trimBST(root->left, L, R);
// 根结点不需要修剪,则修剪左右子树
root->left = trimBST(root->left, L, R);
root->right = trimBST(root->right, L, R);
// 修剪完成,返回根结点
return root;
}
};
作者:@臭咸鱼
转载请注明出处:https://www.cnblogs.com/chouxianyu/
欢迎讨论和交流!