文章目录
- 1. 题目
- 2. 解题
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 6 和 2/ \ /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阿明),一起加油、一起学习进步!