• 博客园logo
  • 会员
  • 众包
  • 新闻
  • 博问
  • 闪存
  • 赞助商
  • HarmonyOS
  • Chat2DB
    • 搜索
      所有博客
    • 搜索
      当前博客
  • 写随笔 我的博客 短消息 简洁模式
    用户头像
    我的博客 我的园子 账号设置 会员中心 简洁模式 ... 退出登录
    注册 登录
THE WAY I AM
博客园    首页    新随笔    联系   管理    订阅  订阅
669. Trim a Binary Search Tree 修剪二叉搜索树 Java

题目要求对一棵二叉搜索树进行规定的修剪:剪去小于x的,大于y的。

我们只需要递归地对树的左子树右子树进行修剪即可。

所有情况如下:

  • 根root的值不在(x,y)内,则判断是小于x还是大于y,分别返回右子树的修剪和左子树的修剪(因为如果root小于x,那么它的左子树必定都小于x,所以我们只需考虑它的右子树。root大于y同理。)
  • root的值在(x,y)内:
    • 对root的左子树进行同样的修剪,并把结果返回,赋给root的左子树,成为root的新左子树
    • 对root的右子树进行同样的修剪,并把结果返回,赋给root的右子树,成为root的新右子树
  • root为空null,说明树的结构不再延续,要返回null
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode trimBST(TreeNode root, int L, int R) {
        if(root == null)
            return null;
        if(root.val < L)
            return trimBST(root.right, L, R);
        if(root.val > R)
            return trimBST(root.left, L, R);
        root.left = trimBST(root.left, L, R);
        root.right = trimBST(root.right, L, R);
        return root;    
    }
}

 

posted on 2019-05-17 11:05  549  阅读(87)  评论(0)    收藏  举报
刷新页面返回顶部
博客园  ©  2004-2025
浙公网安备 33010602011771号 浙ICP备2021040463号-3