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

669. 修剪二叉搜索树

思路

递归法： 

需要思考清楚，如果当前节点<low，那么就返回递归它的右节点，而不是自己取判断，找出来一个合适的节点，这样的话会越想越乱

代码：

TreeNode* trimBST_cursor(TreeNode* root, int low, int high) {
    if (!root) return nullptr;

    if (root->val < low)
    {
        return trimBST_cursor(root->right, low, high);
    }

    if (root->val > high)
    {
        return trimBST_cursor(root->left, low, high);
    }

    if (root->left)
    {
        root->left = trimBST_cursor(root->left, low, high);
    }

    if (root->right)
    {
        root->right = trimBST_cursor(root->right, low, high);
    }

    return root;
}

迭代法：

需要先把当前的root，放到符合标准的区间里，这样不管左子树的右孩子，再怎么大，也不会大过root，所以他也就不会超过high

代码

TreeNode* trimBST(TreeNode* root, int low, int high) {
    if (!root) return nullptr;

    //当前root位于最近的[]范围里
    while (root&&(root->val<low||root->val>high))
    {
        if (root->val < low)
            root = root->right;
        else if (root->val > high)
            root = root->left;
    }

    //让cur = root，开始向左遍历，如果左孩子小于low，那么就一直遍历它的右孩子，直到它的右孩子>low
    auto curNode = root;
    while (curNode)
    {
        while (curNode->left && curNode->left->val < low)
        {
            curNode->left = curNode->left->right;
        }
        //遇到一个左孩子小于的话，那么就把他删掉
        curNode = curNode->left;
    }

    curNode = root;
    while (curNode)
    {
        while (curNode->right && curNode->right->val > high)
        {
            curNode->right = curNode->right->left;
        }

        curNode = curNode->right;
    }

    return root;
}

 108.将有序数组转换为二叉搜索树

比较简单，直接上代码：

TreeNode* sortedArrayToBST(vector<int>& nums)
{
    if (nums.empty()) return nullptr;

    int midIndex = nums.size() / 2;

    TreeNode* root = new TreeNode(nums[midIndex]);

    vector<int> left = vector<int>(nums.begin(), nums.begin() + midIndex);

    vector<int>right = midIndex + 1 < nums.size() ? vector<int>(nums.begin() + midIndex + 1, nums.end()) : vector<int>{};

    root->left = sortedArrayToBST(left);
    root->right = sortedArrayToBST(right);

    return root;
}

 

 538.把二叉搜索树转换为累加树  

思路：

1，中序遍历，然后找到所有的节点，然后从后到前的更改数值

2，定义一个pre，来存储前一个的数值

代码：

void convertBST_cur(TreeNode* root, vector<TreeNode*>& nodes)
{
    if (!root) return ;
    if (root->left) convertBST_cur(root->left, nodes);
    nodes.push_back(root);
    if (root->right) convertBST_cur(root->right, nodes);
}
TreeNode* convertBST(TreeNode* root) {
    if (!root) return nullptr;

    stack<TreeNode*> selected;
    selected.push(root);
    vector<TreeNode*> nodes;
    convertBST_cur(root, nodes);
    
    //如果更改它们的值，那么就会导致链接出现了问题
    for (int i = nodes.size() - 2; i >= 0; i--)
    {
        nodes[i]->val += nodes[i + 1]->val;
    }

    return root;
}