D21|二叉搜索树part2+最近公共祖先(再复习)

530.二叉搜索树的最小绝对差

1.题目
给你一个二叉搜索树的根节点 root ,返回树中任意两不同节点值之间的最小差值 。差值是一个正数,其数值等于两值之差的绝对值。530.二叉搜索树的最小绝对差
2.复习前
BST,中序遍历,但如何写pre和cur呢?
如果不会写,就导出数组,依次比较前后差

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def __init__(self):
        self.res = inf
        self.pre = None
    
    def getMin(self, cur):
        if not cur:
            return
        self.getMin(cur.left)
        if self.pre:
            self.res = (cur.val - self.pre.val) if (cur.val - self.pre.val) < self.res else self.res
        self.pre = cur  # 记录该节点,从而去改变cur
        self.getMin(cur.right)
        
    def getMinimumDifference(self, root: Optional[TreeNode]) -> int:
        # 定义pre和cur,以中序遍历的形式访问节点,并记录最小差值
        # 对于一棵树需要比较两个节点时,建议只对其中一个节点进行遍历,另一个节点赋值改变
        # 如需将两个节点同时传入时,一般是独立的两棵树,如合并二叉树
        self.getMin(root)
        return self.res

3.文章讲解
总结:对于一棵树需要比较两个节点pre cur时,建议只对其中一个节点进行遍历,另一个节点赋值改变,改变的位置即处理根的位置

501.二叉搜索树中的众数

1.题目
给你一个含重复值的二叉搜索树(BST)的根节点 root ,找出并返回 BST 中的所有 众数(即,出现频率最高的元素)。如果树中有不止一个众数,可以按 任意顺序 返回。501.二叉搜索树中的众数
2.实现
实现时,对于末位众数的情况出现错误,仔细想想!

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def __init__(self):
        self.cnt = -inf  # 记录最大频数
        self.res = []  # 记录结果
        self.tmp = 1  # 记录目前的个数
        self.pre = None
    
    def getMin(self, cur):
        if not cur:
            return
        self.getMin(cur.left)
        if self.pre and self.pre.val == cur.val:
            self.tmp += 1
        else:
            self.tmp = 1
        if self.cnt < self.tmp:  # 改变之后就得去更新结果,否则如果最后一种情况是最大众数,由于下一个cur==None,则不会再执行
            self.res.clear()
            self.cnt = self.tmp
            self.res.append(cur.val)
        elif self.cnt == self.tmp:
            self.res.append(cur.val)
        self.pre = cur  # 记录该节点,从而去改变cur
        self.getMin(cur.right)

    def findMode(self, root: Optional[TreeNode]) -> List[int]:

        self.getMin(root)
        return self.res

3.文章讲解
如果是比赛的话,建议学会普通树下用hash表去记录值从而求众数的情况!(代码简洁,且时间复杂度没有高太多)

236. 二叉树的最近公共祖先

1.题目
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。 236. 二叉树的最近公共祖先
2.思路:
开始用两种状态去标记是否遍历了p q,都为true时则改变res,但会出现自身为祖先或者根为祖先冲突的情况

class Solution:
    def __init__(self):
        self.res = None
        self.s1 = False
        self.s2 = False

    def lowest(self, root, p, q):
        if not root or (self.s1 and self.s2):
            return
        self.lowestCommonAncestor(root.left, p, q)
        self.lowestCommonAncestor(root.right, p, q)
        # 这种方法只是说明找到了节点,但会出现以下情况:
        # 1.如p=5 q=4时p为公共祖先,如果改变s1 和s2之后更新res,则会导致5 1的公共祖先变为1
        # 2.如p=5 q=4时p为公共祖先,如果改变s1 和s2,直接return,则会找不到5为公共祖先,而是5的根节点为祖先
        if root == p:
            self.s1 = True
            return
        if root == q:
            self.s2 = True
            return
        if self.s1 and self.s2:
            self.res = root  

    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        # 采用后序遍历,用两种状态去记录是否遍历过两个节点,一旦都为true则返回根节点   不太行!

3.文章讲解
思路:
1.结合图好好理解如何去写遍历p q后的情况
公共祖先返回节点情况
2.为什么需要写有返回值的函数
很好的解决了上面那种思路遇到的问题,这里可以根据left和right来去返回不同的root

    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        if not root:
            return root
        left = self.lowestCommonAncestor(root.left, p, q)
        right = self.lowestCommonAncestor(root.right, p, q)
        if root == p or root == q:
            return root
        if left and right:
            return root
        elif left and not right:
            return left
        elif right and not left:
            return right
        else:
            return None

总结:
1.重刷好好理解遍历过程
2.以后对于需要确定返回值的情况得画个图,主要是搞清楚遍历顺序

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值