前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 776. 拆分二叉搜索树(DFS)*

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

作者头像
Michael阿明
发布2021-02-19 10:09:20
5350
发布2021-02-19 10:09:20
举报
文章被收录于专栏:Michael阿明学习之路

文章目录

1. 题目

给你一棵二叉搜索树(BST)、它的根结点 root 以及目标值 V。

请将该树按要求拆分为两个子树:其中一个子树结点的值都必须小于等于给定的目标值 V;另一个子树结点的值都必须大于目标值 V;树中并非一定要存在值为 V 的结点。

除此之外,树中大部分结构都需要保留,也就是说原始树中父节点 P 的任意子节点 C,假如拆分后它们仍在同一个子树中,那么结点 P 应仍为 C 的子结点。

你需要返回拆分后两个子树的根结点 TreeNode,顺序随意。

代码语言:javascript
复制
示例:
输入: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 = 子树的小节点
代码语言:javascript
复制
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

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2020/07/22 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 文章目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档