写代码的第十八天,二叉树最后一天!!!加油呀!!!
现在的重点还是在递归上,迭代我先放一放了,二刷的时候再重点看迭代。
今天居然没有写错的代码,稀奇了,整理思路还是有用。
669. 修剪二叉搜索树
思路(递归)
找到需要删除的结点,如果该结点的值小于最小范围值,那么递归遍历其右子树,把右子树中符合范围的子树保留返回给待删除结点的父结点;如果该结点的值大于最大范围值,那么递归其左子树,把左子树中符合范围的子树保留返回给待删除结点的父结点;现在处理完根结点了,然后就需要处理左右子树,那么就需要递归的从根结点开始遍历,先遍历左子树,在遍历右子树。
解决问题1:参数和返回值?参数需要当前树的根结点以及修剪的范围;返回值是新树的根结点。
解决问题2:终止条件?我们要遍历完整颗树,所以终止条件是if root= =None,返回None。
解决问题3:单次递归逻辑?按照中左右的方式遍历,首先处理中间结点,如果root.val<left,那么trimBST(root.right,left,right),然后return 上面的式子;如果root.val<right,那么trimBST(root.left,left,right),然后return上面的式子。接下来处理左右子树,还是trimBST(root.left,left,right)和trimBST(root.right,left,right)。
正确代码:
class Solution:
def trimBST(self, root: Optional[TreeNode], low: int, high: int) -> Optional[TreeNode]:
if root == None:
return None
if root.val < low:
right = self.trimBST(root.right,low,high)
return right
if root.val > high:
left = self.trimBST(root.left,low,high)
return left
root.left = self.trimBST(root.left,low,high)
root.right = self.trimBST(root.right,low,high)
return root
108.将有序数组转换为二叉搜索树
思路(递归)
这个题和之前的中后序构造二叉树这类题很相似。采用的是一种方法,都是找到结点然后切割左右两部分,左边递归,右边递归。
解决问题1:参数和返回值?参数需要当前树的根结点数组的左右范围;返回值是新树的根结点。有点像二分查找诶
解决问题2:终止条件?我们是根据有序数组来构建二叉树,所以终止条件和树中的结点没关系,应该是当数组中无数据了这时候终止。那条件应该怎么写呢?我们在函数中传入了数组的左和右然后找mid中结点,两边递归,这个时候左右就很像二分查找中的left指针和right指针,回想一下在二分查找中循环停止条件是left<right,这个条件产生的原因是我们采用的左闭右开的区间,所以本题中也可以用左闭右开的情况,当left大于right的时候停止递归。
解决问题3:单次递归逻辑?首先找到根结点,也就是数组中的中间结点,当数组长度是奇数的时候直接用数组中间的值,当数组长度是偶数的时候我这里设置根结点为(left+right)/2这个位置的值。然后递归遍历左子树和右子树。
正确代码:
class Solution:
def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
return self.traversel(nums,0,len(nums))
def traversel(self,nums,left,right):
if left >= right:
return None
mid = (left + right) // 2
root = TreeNode(nums[mid])
root.left = self.traversel(nums,left,mid)
root.right = self.traversel(nums,mid+1,right)
return root
538.把二叉搜索树转换为累加树
思路(递归)
这个题要求的是累加树,把比这个结点值大的那些都加起来作为当前结点的值,这是一颗二叉搜索树,所以本身就是有序的,是从小到大的,按照左中右顺序,但是现在要从大到小的累加,那么我们可以反向一下,右中左这个顺序就是从大到小了。每一次都需要累加一个结点所以我们可以设置两个指针,一个遍历整颗树,一个在这个指针之前pre用来记录前一个结点的值,这样每次就不需要遍历整颗树重新相加了,只需要每次当前结点加pre就可以了。
解决问题1:参数和返回值?参数是跟结点,返回值也是新的树根结点。
解决问题2:终止条件?当到叶子结点的时候也就是root==空的时候停止。
解决问题3:单次递归逻辑?按照右中左的逻辑遍历,设置全局变量pre=0,为了累加计数,设置cur指针从最右侧开始,每次累加更新值
正确代码:
class Solution:
def __init__(self):
self.pre = 0
def convertBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
root = self.traversal(root)
return root
def traversal(self,cur):
if cur == None:
return None
cur.right = self.traversal(cur.right)
cur.val += self.pre
self.pre = cur.val
cur.left = self.traversal(cur.left)
return cur