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

萌帅帅小明

已于 2024-02-01 16:09:17 修改

阅读量369

点赞数 10

文章标签：算法 leetcode 职场和发展 c# 数据结构

于 2024-02-01 16:08:59 首次发布

本文链接：https://blog.csdn.net/ShengHuoWeiYang/article/details/135976023

版权

669.修剪二叉搜索树

public class Solution {
    public TreeNode TrimBST(TreeNode root, int low, int high) {
        if(root==null)
        {
            return null;
        }
        if(root.val<low)
        {
            var right=TrimBST(root.right,low,high);
            return right;
        }
        if(root.val>high)
        {
            var left=TrimBST(root.left,low,high);
            return left;
        }
        root.left=TrimBST(root.left,low,high);
        root.right=TrimBST(root.right,low,high);
        return root;
    }
}

首先如果根节点为空则直接返回空，如果根节点的值小于Low，由于是二叉搜索树所以不用往左继续去寻找，那就往右寻找一个满足条件的节点，同理节点大于High值则往左寻找第一个满足条件的节点，处理完一个的逻辑接着就去左右子树递归调用，最终返回根节点。

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

public class Solution {
    public TreeNode SortedArrayToBST(int[] nums) {
        return Traversal(nums,0,nums.Length-1);
    }
    public TreeNode Traversal(int[]nums,int left,int right)
    {
        if(left>right)
        {
            return null;
        }
        var mid=(left+right)/2;
        var root=new TreeNode(nums[mid]);
        root.left=Traversal(nums,left,mid-1);
        root.right=Traversal(nums,mid+1,right);
        return root;
    }
}

因为要是平衡树所以直接找数组中点作为搜索树的根节点，两边才会平衡，找到之后分别去中点左右两区区间递归即可，注意区间是左闭右开还是左闭右闭！

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

public class Solution {
     int pre=0;
    public TreeNode ConvertBST(TreeNode root) {
        pre=0;
        Traversal(root);
        return root;

    }
    public void Traversal(TreeNode cur)
    {
        if(cur==null)
        {
            return;
        }
        Traversal(cur.right);
        cur.val+=pre;
        pre=cur.val;
        Traversal(cur.left);
    }
}

题目不容易读明白，但读明白之后就相当于取一个有序数组从后往前叠加的操作。所以我们需要用到双指针Cur和Pre指针，Pre首先等于0，如果当前节点为空直接就返回，递归顺序采取右中左，没个节点都加上前一个节点的值Pre加完后就变成当前节点的值准备赋值给下一个，递归完成后返回根节点即可。