【Leetcode】776. Split BST

题目地址：

https://leetcode.com/problems/split-bst/

给定一棵二叉搜索树，再给定一个数值$x$，要求将该BST分裂成两棵BST，第一个小于等于$x$，另一个大于$x$。在结果中，要求尽量保持原有的结构，除了必须分裂的地方，其余节点的父子关系应该保持原样。

思路是分治。设 f [ r t ] = { l , r } f[rt]=\{l,r\} f[rt]={l,r}，$rt$是当前树根，$l$是分裂出来的小于$x$的BST树根，$r$是分裂出来的大于$x$的BST树根。如果$x\ge rt$，那左子树应该作为一个整体挂在$rt$的左边，而右子树应该去分裂，设右子树为$rt.r$递归分裂出 f [ r t . r ] = { l 2 , r 2 } f[rt.r]=\{l_2,r_2\} f[rt.r]={l2​,r2​}，由于$l_2$都是小于等于$x$而大于树根的，所以其应该接到$rt.r$处形成第一个BST，而$r_2$是大于$x$的，它单独作为第二个BST，所以就得到了结果的两部分；如果$x<rt$，那右子树应该作为一个整体挂在$rt$的右边，而左子树应该去分裂，设左子树为$rt.l$递归分裂出 f [ r t . l ] = { l 1 , r 1 } f[rt.l]=\{l_1,r_1\} f[rt.l]={l1​,r1​}，由于$r_1$都是大于$x$而小于树根的，所以其应该接到$rt.l$处形成第二个BST，而$l_1$小于等于$x$的，它单独作为第一个BST，所以就得到了结果的两部分。代码如下：

class Solution {
 public:
  vector<TreeNode*> splitBST(TreeNode* root, int t) {
    if (!root) return {nullptr, nullptr};
    if (root->val <= t) {
      auto res = splitBST(root->right, t);
      root->right = res[0];
      return {root, res[1]};
    } else {
      auto res = splitBST(root->left, t);
      root->left = res[1];
      return {res[0], root};
    }
  }
};

时间复杂度$O(n)$，空间$O(\log n)$。