二叉搜索树最近节点查询python

 给你一个 二叉搜索树 的根节点 root ，和一个由正整数组成、长度为 n 的数组 queries 。

请你找出一个长度为 n 的 二维 答案数组 answer ，其中 answer[i] = [mini, maxi] ：

• mini 是树中小于等于 queries[i] 的 最大值 。如果不存在这样的值，则使用 -1 代替。
• maxi 是树中大于等于 queries[i] 的 最小值 。如果不存在这样的值，则使用 -1 代替。

返回数组 answer 。

class Solution:
    def closestNodes(self, root: Optional[TreeNode], queries: List[int]) -> List[List[int]]:

        def inorder(root):
            if not root: return
            inorder(root.left)
            sortedTree.append(root.val)
            inorder(root.right)
        
        def bisect(target):
            i,j = 0,len(sortedTree)-1
            bigger,lesser = -1,-1
            while i <= j:
                m = i + (j - i) // 2
                if sortedTree[m] > target:
                    bigger = sortedTree[m]
                    j = m - 1
                elif sortedTree[m] < target:
                    lesser = sortedTree[m]
                    i = m + 1
                else:
                    return [sortedTree[m],sortedTree[m]]
            
            return [lesser,bigger]

        sortedTree = []
        inorder(root)
        ans = []
        for v in queries:
            ans.append(bisect(v))

        return ans

题目当中未说明树是平衡树，所以如果当树退化为链表，用树的性质对值进行查询就会超时（O(n)），这样对每一个询问数组内的元素都要进行一次O(n)总时间复杂度为O(qn)。

更快的做法是，通过中序遍历将树转化为排序数组，二分查找即可。

代码在经典的二分查找上进行了修改，使其能够返回题目所要求的答案，无需进行额外处理，但是时间上会慢一些。时间复杂度O(qlogn)