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

669.修剪二叉搜索树

public class Solution {
    public TreeNode TrimBST(TreeNode root, int low, int high) {
        if(root==null)
        {
            return null;
        }
        if(root.valhigh)
        {
            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加完后就变成当前节点的值准备赋值给下一个,递归完成后返回根节点即可。

你可能感兴趣的:(算法,leetcode,职场和发展,c#,数据结构)