代码随想录算法训练营第二十三天| 669. 修剪二叉搜索树 108.将有序数组转换为二叉搜索树 538.把二叉搜索树转换为累加树

题目链接：669. 修剪二叉搜索树

• 确定终止条件

修剪的操作并不是在终止条件上进行的，所以就是遇到空节点返回就可以了。

if (root == nullptr ) return nullptr;

1

• 确定单层递归的逻辑

如果root（当前节点）的元素小于low的数值，那么应该递归右子树，并返回右子树符合条件的头结点。

代码如下：

if (root->val < low) 
{
    TreeNode* right = trimBST(root->right, low, high); 
    return right;
}

如果root(当前节点)的元素大于high的，那么应该递归左子树，并返回左子树符合条件的头结点。

代码如下：

if (root->val > high) {
    TreeNode* left = trimBST(root->left, low, high);
    return left;
}

整体代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high)
    { if(root==nullptr) return root;
    if(root->val<low)
     {
        TreeNode* right=trimBST(root->right,low,high);
        return right;
     }
    if(root->val>high)
     {
        TreeNode* left=trimBST(root->left,low,high);
        return left;
     }  
    root->left=trimBST(root->left,low,high);
    root->right=trimBST(root->right,low,high);
    return root;
    }
};

题目链接：108.将有序数组转换为二叉搜索树

采用递归法，定义左闭右闭的区间

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
private:
    TreeNode* traversal(vector<int>& nums,int left,int right) 
    {
        if(left>right) return nullptr;
        int mid=(left+right)/2;
        TreeNode* root=new TreeNode(nums[mid]);
        root->left=traversal(nums,left,mid-1);
        root->right=traversal(nums,mid+1,right);
        return root;
    }   
public:
    TreeNode* sortedArrayToBST(vector<int>& nums) 
    {
        
        return traversal(nums,0,nums.size()-1);
    }
};

题目链接：538.把二叉搜索树转换为累加树

累加树，可以看做求从后到前的累加数组，采用逆中序遍历

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
private:
 int pre=0;
 void traversal(TreeNode* root)
 {
     if(root==nullptr) return;
     traversal(root->right);
     root->val+=pre;
     pre=root->val;
     traversal(root->left);
 }
public:
    TreeNode* convertBST(TreeNode* root) 
    {   
        traversal(root);
        return root;

    }
};