面试常见算法题目

1 TOP K

"""35 最小的k个数
TOPK问题： 一般建立大根堆，  如果是最大的就建立小根堆
快排也可以实现
"""
# 快排
def p( L, l, r):
    temp, i = L[r], l - 1
    for j in range(l, r):
        if L[j] <= temp:
            i += 1
            L[i], L[j] = L[j], L[i]
    L[i + 1], L[r] = L[r], L[i + 1]
    return i + 1

def GetLeastNumbers_Solution( tinput, k):
    if k == 0 or k > len(tinput):
        return []
    else:
        flag = p(tinput, 0, len(tinput) - 1)  #拆分点
        while flag != k - 1:
            if flag > k - 1:
                flag = p(tinput, 0, flag - 1)
            if flag < k - 1:
                flag = p(tinput, flag + 1, len(tinput) - 1)
        return sorted(tinput[:flag + 1])
#堆排序（TOP） 这是找最大的。
class Solution:
    def __init__(self):
        pass
    def Heap(self, L, i, size):  #小根堆维护
        root = i
        left , right = 2*i+1, 2*i+2
        if left < size and L[root]> L[left]:
            root = left
        if right < size and L[root]> L[right]:
            root = right
        if root != i:
            L[root],L[i] = L[i],L[root]
            self.Heap(L,root,size)
        return L
    def buildHeap(self, L):
        for i in range(len(L)//2,-1,-1):
            self.Heap(L,i,len(L))
        return L
    def TopK(self, L, k):
        result = L[:k]
        result = self.buildHeap(result)
        for i in range(k,len(L)):
            if L[i] > result[0]:
                result[0] = L[i]
                self.Heap(result, 0, k)
        return result
""""""