代码随想录day23 二叉岁终章

669. 修剪二叉搜索树

题目

给定一个二叉搜索树,同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树,使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点,所以结果应当返回修剪好的二叉搜索树的新的根节点。

代码随想录day23 二叉岁终章_第1张图片

思考

这题有个非常大的坑,就是当root小于low或者high时,很多人会把root变为nullptr,但其实在root的左右子树里可能还会有值满足条件,所以不能直接变为nullptr,要一直遍历可能存在的值,即当root小于low时,递归遍历它的右子树,当大于high时,递归遍历它的左子树,并且最后返回它搜索到符合条件的最后值

代码

class Solution {

public:

    TreeNode* trimBST(TreeNode* root, int low, int high) {

        if(root == nullptr) return nullptr;

        if(root->val > high) {

            TreeNode* left= trimBST(root->left, low, high);

            return left;

        }

        else if(root->val < low) {

            TreeNode* right = trimBST(root->right, low, high);

            return right;

        }

        root->left = trimBST(root->left, low, high);

        root->right = trimBST(root->right, low, high);

        return root;

    }

};

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

题目

将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。

本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

示例:

代码随想录day23 二叉岁终章_第2张图片

思考

在做完之前给出前序中序数组再组成一个二叉树的题目后,发现类似的题目最关键的一点就是找到根结点,这题的根结点就是在数组的中间,然后递归遍历左数组,递归遍历右数组即可,需要注意的是,每次递归的mid是新数组的mid,所以不需要用left再加一次

代码

class Solution {

public:

    TreeNode* traversal(vector& nums, int left, int right) {

        if(left > right) return nullptr;

        int mid = left + (right - left)/2;

        TreeNode* node = new TreeNode(nums[mid]);

        node->left = traversal(nums, left, mid - 1);

        node->right = traversal(nums, mid+1, right);

        return node;

    }

    TreeNode* sortedArrayToBST(vector& nums) {

        TreeNode* root = traversal(nums, 0, nums.size()-1);

        return root;

    }

};

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

题目

给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

提醒一下,二叉搜索树满足下列约束条件:

节点的左子树仅包含键 小于 节点键的节点。 节点的右子树仅包含键 大于 节点键的节点。 左右子树也必须是二叉搜索树。

示例 1:

代码随想录day23 二叉岁终章_第3张图片

  • 输入:[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
  • 输出:[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

思考

首先解释题目意思:这题其实就是把每个结点都加上其前结点一起累加的数,从图中可以看出,顺序是右中左(其实我也没明白为啥是右中左,感觉左中右也行啊);本题运用的方法就是双指针,这里比较特别的是pre指针是int型,cur指针是TreeNode,因为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;

    }

};

你可能感兴趣的:(算法,b树,数据结构)