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

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

669. 修剪二叉搜索树

题目链接：669. 修剪二叉搜索树

递归法

思路：本题易错点在于，当某个节点不在给定区间里时，例如小于左区间，不能单纯将该节点代表的子树直接删除，因为该节点的右孩子可能满足区间条件，如下图所示：

class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if (root == null) return null;
        // 如果当前节点的元素小于 low，那么应该递归右子树，并返回右子树符合条件的头结点。
        if (root.val < low) {
            TreeNode right = trimBST(root.right, low, high);
            return right;
        }
        // 如果当前节点的元素大于 high，那么应该递归左子树，并返回左子树符合条件的头结点。
        if (root.val > high) {
            TreeNode left = trimBST(root.left, low, high);
            return left;
        }
        // 将下一层处理完左子树的结果赋给 root.left，处理完右子树的结果赋给root.right。
        root.left = trimBST(root.left, low, high);
        root.right = trimBST(root.right, low, high);
        return root;
    }
}

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

题目链接：108.将有序数组转换为二叉搜索树

递归法

思路：本题本质就是寻找分割点，分割点作为当前节点，然后递归左区间和右区间。如果数组长度为偶数，中间节点有两个，取哪一个？ 取哪一个都可以，只不过构成了不同的平衡二叉搜索树。本题选择左闭右闭区间。

class Solution {
	public TreeNode sortedArrayToBST(int[] nums) {
		TreeNode root = traversal(nums, 0, nums.length - 1);
		return root;
	}

	// 左闭右闭区间[left, right]
	private TreeNode traversal(int[] nums, int left, int right) {
		if (left > right) return null;

		int mid = left + ((right - left) >> 1); // 防止整数溢出
		TreeNode root = new TreeNode(nums[mid]);
		root.left = traversal(nums, left, mid - 1);
		root.right = traversal(nums, mid + 1, right);
		return root;
	}
}

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

题目链接：538.把二叉搜索树转换为累加树

双指针法 + 递归法

思路：从题目可以看出，本题对树的遍历顺序是右中左，即反中序遍历。利用双指针进行累加。

class Solution {
    private int pre;
    public TreeNode convertBST(TreeNode root) {
        pre = 0;
        traversal(root);
        return root;
    }
    public void traversal(TreeNode root) {
        if (root == null) return;
        traversal(root.right);
        pre += root.val;
        root.val = pre;
        traversal(root.left);
    }
}

二叉树总结

代码随想录——二叉树完结篇
二叉树专题汇总：