LeetCode 776. 拆分二叉搜索树(DFS)*

文章目录

1. 题目

给你一棵二叉搜索树(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      62
            / \           /
           5   7         1
 

提示:

二叉搜索树节点个数不超过 50 
二叉搜索树始终是有效的,并且每个节点的值都不相同

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/split-bst
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

  • 函数返回值(小值根节点,大值根节点)
  • root 左边存在答案(其val > v),root->left = 子树的大节点,这样才能分开
  • root 右边存在答案(其val <= v),root->right = 子树的小节点
class Solution {
public:
    vector<TreeNode*> splitBST(TreeNode* root, int V) {
        if(!root)
            return {NULL, NULL};
        if(root->val > V)
        {
            auto l = splitBST(root->left, V);
            root->left = l[1];
            return {l[0], root};
        }
        else
        {
            auto r = splitBST(root->right, V);
            root->right = r[0];
            return {root, r[1]};
        }
    }
};

4 ms 9 MB


我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!
Michael阿明

猜你喜欢

转载自blog.csdn.net/qq_21201267/article/details/107520331