LeetCode——拆分二叉搜索树

Q：给你一棵二叉搜索树（BST）、它的根结点 root 以及目标值 V。
请将该树按要求拆分为两个子树：其中一个子树结点的值都必须小于等于给定的目标值 V；另一个子树结点的值都必须大于目标值 V；树中并非一定要存在值为 V 的结点。
除此之外，树中大部分结构都需要保留，也就是说原始树中父节点 P 的任意子节点 C，假如拆分后它们仍在同一个子树中，那么结点 P 应仍为 C 的子结点。
你需要返回拆分后两个子树的根结点 TreeNode，顺序随意。

示例：
输入：root = [4,2,6,1,3,5,7], V = 2
输出：[[2,1],[4,3,6,null,null,5,7]]
解释：
注意根结点 output[0] 和 output[1] 都是 TreeNode 对象，不是数组。
给定的树 [4,2,6,1,3,5,7] 可化为如下示意图：

      4
    /   
  2      6
 /     / 
1   3  5   7

输出的示意图如下：

      4
    /   
  3      6       和    2
        /            /
       5   7         1

A：
根节点要么在第一棵子树中，要么在第二棵子树中。假设根节点在第一棵子树中，那么这棵树的左子树一定在第一棵子树中，右子树中部分节点在第一棵子树，部分在第二棵子树中。

bns = split(root.right) 为分割树的结果。bns[0] 和 bns[1] 在分割之后依然还是二叉搜索树，其中 bns[0] 在第一棵子树中，bns[1] 为第二棵子树。

如上图所示，需要将 split(root.right) 的结果合并到跟节点所在的子树中。

class Solution {
    public TreeNode[] splitBST(TreeNode root, int V) {
        if (root == null)
            return new TreeNode[]{null, null};
        else if (root.val <= V) {
            TreeNode[] bns = splitBST(root.right, V);
            root.right = bns[0];
            bns[0] = root;
            return bns;
        } else {
            TreeNode[] bns = splitBST(root.left, V);
            root.left = bns[1];
            bns[1] = root;
            return bns;
        }
    }
}