type TreeNode struct {
Val int
Left *TreeNode
Right *TreeNode
}
//669. 修剪二叉搜索树
//本题与450. 删除二叉搜索树中的节点 相似，不同之处在于本题删除不止一个节点，450不需要遍历全部二叉树，但本题需要全部遍历，而且要随时调整二叉树结构
func trimBST(root *TreeNode, low int, high int) *TreeNode {
if root == nil {
return root
}
//找到要删除的左子树
if root.Val < low {
//因为不确定左子树是否还要删除，因此要不断向右遍历，因为是二叉树搜索树，当前节点如果比low小，当前节点左子树一定比low小，因此直接舍弃，不用遍历
right := trimBST(root.Right, low, high)
return right
}
if root.Val > high {
//同样因为不确定右子树是否还要删除，因此要不断向左遍历，因为是二叉树搜索树，当前节点如果比high 大，当前节点右子树一定比high 大，因此直接舍弃，不用遍历
life := trimBST(root.Left, low, high)
return life
}
root.Left = trimBST(root.Left, low, high)
root.Right = trimBST(root.Right, low, high)
return root
}
//108. 将有序数组转换为二叉搜索树
//因为是有序数组，让中间节点作为根据节点，然后进行前序遍历中左右
func sortedArrayToBST(nums []int) *TreeNode {
if len(nums) == 0 {
return nil
}
//寻找中间值索引
mid := len(nums) / 2
root := &TreeNode{Val: nums[mid]} //中
root.Left = sortedArrayToBST(nums[:mid]) //左
root.Right = sortedArrayToBST(nums[mid+1:]) //右
return root
}
//538. 把二叉搜索树转换为累加树
func convertBST(root *TreeNode) *TreeNode {
if root == nil {
return nil
}
prev := 0
var traversal func(tree *TreeNode)
traversal = func(tree *TreeNode) {
if tree == nil {
return
}
// 中序遍历是左中右，但我们要先取最大值，所以要逆中序遍历，即右中左
traversal(tree.Right) // 右
tree.Val = tree.Val + prev //中
prev = tree.Val //把当前值传给前一个指针
traversal(tree.Left) //左
}
traversal(root)
return root
}
)
