代码随想录算法训练营Day23 | 二叉树part09

原创已于 2024-01-18 11:33:20 修改 · 580 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #c++ #leetcode #数据结构

于 2024-01-18 11:32:54 首次发布

文章讲述了如何在LeetCode中解决关于二叉搜索树的三个问题：修剪BST以满足区间条件、将有序数组转换为BST以及将BST转换为累加树。涉及递归、区间划分和双指针技巧的应用。

669. 修剪二叉搜索树

leetcode链接
 代码随想录链接
一刷状态：未通过

思路

递归思路比较巧妙，先对不符合区间的节点继续遍历修剪，使用左右孩子去接裁剪后返回的子树。

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if(root==nullptr) return nullptr;
        // 如果节点的值小于区间，则返回以root->right为根节点的裁剪后的子树
        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;
    }
};

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

leetcode链接
 代码随想录链接
一刷状态：通过

思路

构造二叉树的思路，分左右区间分别构造，注意区间的定义，本题使用左闭右闭。
递归的终止条件为非法区间。

class Solution {
public:
    TreeNode* traversal(vector<int>& nums, int left, int right)     // 左闭右闭
    {
        if(left>right) return nullptr;

        int mid = left + (right-left)/2;
        TreeNode* root = new TreeNode(nums[mid]);
        root->left = traversal(nums, left, mid-1);
        root->right = traversal(nums, mid+1, right);
        return root;
    }

    TreeNode* sortedArrayToBST(vector<int>& nums) {
        return traversal(nums, 0, nums.size()-1);
    }
};

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

leetcode链接
 代码随想录链接
一刷状态：通过

思路

右中左遍历，使用双指针法，从后往前累加，使用pre记录上一个节点的值，加到当前节点。

class Solution {
public:
    int pre = 0;
    void traversal(TreeNode* cur)
    {
        if(cur==nullptr) return;
        traversal(cur->right);

        cur->val += pre;
        pre = cur->val;
        
        traversal(cur->left);
    }

    TreeNode* convertBST(TreeNode* root) {
        traversal(root);
        return root;

    }
};