day 23 第六章 二叉树part09

● 669. 修剪二叉搜索树
● 108.将有序数组转换为二叉搜索树
● 538.把二叉搜索树转换为累加树
● 669. 修剪二叉搜索树

class Solution_669 {
 public:
  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) {}
  };
  /*
   * 修剪指定区间，按中序遍历，很容易把区间内的右孩子节点剪掉，所以在剪掉中间节点时，将右孩子节点挂到上一节点的左孩子节点
输入参数为当前节点，返回值为修剪后的节点
终止条件为当前节点是否为空
单层递归逻辑：
    判断终止条件，为真返回空
    若当前节点值大于high，遍历左子节点，返回得到的结果
    若当前节点值小于low，遍历右子节点，返回得到的结果
    处理当前节点的下一层节点，以便接住下层向上返回的节点
    返回当前节点
   */
  TreeNode *get(TreeNode *root, int low, int high){
      if(!root)
          return nullptr;

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

      //处理当前节点的下一层节点，以便接住下层向上返回的节点
      root->left = get(root->left, low, high);
      root->right = get(root->right, low, high);

      return root;
  }

  TreeNode *trimBST(TreeNode *root, int low, int high) {
      return get(root, low, high);
  }
};

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

/*
 * 108. 将有序数组转换为二叉搜索树
简单
给你一个整数数组 nums ，其中元素已经按 升序 排列，请你将其转换为一棵平衡二叉搜索树。
 */

class Solution_108 {
 public:
  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) {}
  };
/*
 * 关键时平衡，左右节点数要相等
 输入参数为递增数组，左右区间下标，返回值为中间节点
终止条件为左下标大于右下标
单层递归逻辑：
    判断终止条件，为真返回空
    取左右下标平均值mid，构造当前节点
    遍历当前节点左子节点，应该是[左，mid - 1]区间内
    遍历当前节点左子节点，应该是[mid + 1， 右]区间内
    返回当前节点
 */
  TreeNode *get(vector<int> &nums, int left_index, int right_index){
      if(left_index > right_index)
          return nullptr;
      int mid = left_index + ((right_index - left_index) / 2);
      TreeNode *node = new TreeNode(nums[mid]);
      node->left = get(nums, left_index, mid - 1);
      node->right = get(nums, mid + 1, right_index);
      return node;
  }
  TreeNode *sortedArrayToBST(vector<int> &nums) {
      return get(nums, 0, nums.size() - 1);
  }
};

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

/*
 * 538. 把二叉搜索树转换为累加树
中等
给出二叉 搜索 树的根节点，该树的节点值各不相同，
 请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和
提醒一下，二叉搜索树满足下列约束条件：
节点的左子树仅包含键 小于 节点键的节点。
节点的右子树仅包含键 大于 节点键的节点。
左右子树也必须是二叉搜索树。
 */

class Solution_538 {
 public:
  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) {}
  };

  TreeNode* pre;
  void get(TreeNode *root) {
      if(!root)
          return;
      get(root->right);

      if(pre){
          root->val += pre->val;
      }
      pre = root;

      get(root->left);
  }
  
  TreeNode *convertBST(TreeNode *root) {
      get(root);
      return root;
  }
};