代码随想录算法训练营第23天|● 669. 修剪二叉搜索树 ● 108.将有序数组转换为二叉搜索树 ● 538.把二叉搜索树转换为累加树 ● 总结篇

修剪二叉树

题目链接:

669. 修剪二叉搜索树 - 力扣(LeetCode)

首先判断当前节点是否为空,若为空则返回null;然后判断当前节点的值是否小于low或大于high,分别对应着需要删除左子树或右子树的节点;最后更新当前节点的左子树和右子树,并返回修剪后的当前节点

代码:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 * int val;
 * TreeNode left;
 * TreeNode right;
 * TreeNode() {}
 * TreeNode(int val) { this.val = val; }
 * TreeNode(int val, TreeNode left, TreeNode right) {
 * this.val = val;
 * this.left = left;
 * this.right = right;
 * }
 * }
 */
class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if (root == null) {
            return null;
        }
        if (root.val < low) {
            return trimBST(root.right, low, high);
        }
        if (root.val > high) {
            return trimBST(root.left, low, high);
        }
        // root在[low,high]范围内
        root.left = trimBST(root.left, low, high);
        root.right = trimBST(root.right, low, high);
        return root;
    }
}

将有序数组转换为二叉搜索树 

题目链接

108. 将有序数组转换为二叉搜索树 - 力扣(LeetCode)


 

思路:

利用二叉搜索树的特性,选数组中间位置为当前节点,然后继续向左向右递归

代码:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
     public TreeNode f(int[] nums,int left,int right){
         if(left>right){
             return null;
         }
         int mid=left+(right-left)/2;
         TreeNode root=new TreeNode(nums[mid]);
         root.left=f(nums,left,mid-1);
         root.right=f(nums,mid+1,right);
         return root;
    }
    public TreeNode sortedArrayToBST(int[] nums) {
     TreeNode node = f(nums, 0, nums.length - 1);
        return node;
    }
}

把二叉搜索树转换为累加树 

题目链接:

538. 把二叉搜索树转换为累加树 - 力扣(LeetCode)

思路:

对二叉树进行反中序遍历,对节点数值进行累加

首先,它定义了一个变量pre,用于保存路径上的节点值之和。
如果当前节点为null,表示遍历到叶子节点的子节点,直接返回。
递归调用f函数处理右子树。
更新pre值为当前节点值与pre的和。
将当前节点值更新为pre。
递归调用f函数处理左子树。

代码:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 * int val;
 * TreeNode left;
 * TreeNode right;
 * TreeNode() {}
 * TreeNode(int val) { this.val = val; }
 * TreeNode(int val, TreeNode left, TreeNode right) {
 * this.val = val;
 * this.left = left;
 * this.right = right;
 * }
 * }
 */
class Solution {

    public int sum;

    public void f(TreeNode root) {
        if (root == null) {
            return;
        }
        f(root.right);
        sum += root.val;
        root.val = sum;
        f(root.left);
    }

    public TreeNode convertBST(TreeNode root) {
        sum=0;
        f(root);
        return root;
    }
}

  • 7
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值