这道题主要考察排序算法,之前用快排写的,发现测试用例改了之后没办法通过了。故改用堆排序

堆排序由两部分构成:

1.heapify用来调整堆的结构(使堆这种特殊的完全二叉树中的父节点大于两个孩子节点,但两个孩子节点之间没有大小要求)

2.heapsort部分用来将堆的首元素与末尾元素交换,然后再调用heapify使除了末尾元素之外的n-1个元素来继续形成大顶堆,然后再交换首尾元素,再调整使剩下n-2个元素形成堆,循环往复直至堆内只剩下最后一个元素。

根据堆排序的性质,最好最坏平均时间复杂度都为O(nlogn),且不需要额外开辟空间存储原数组,故空间复杂度O(1)

    def findKth(self , a: List[int], n: int, K: int) -> int:
        def heapify(a, n, i):
            largest = i
            lson = 2 * i + 1
            rson = 2 * i + 2
            if lson < n and a[lson] > a[largest] :
                largest = lson
            if rson < n and a[rson] > a[largest]:
                largest = rson
            if largest != i:
                a[largest], a[i] = a[i], a[largest]
                heapify(a, n, largest)
        def heapsort(a, n):
#            建大顶堆
            for i in range(len(a)//2 - 1, -1 ,-1):
                heapify(a, n, i)
#            排序
            for i in range(len(a) - 1, 0, -1):                       
                a[0], a[i] = a[i], a[0]
                heapify(a, i, 0)
        heapsort(a, n)
        return a[-K]