topK问题

问题：

现在有n个数，设计算法得到前k大的数。（k<n）

解决思路：

(1) 排序后切片 O(nlogn)
(2) 冒泡 O(kn) 选择、插入
(3) 堆排序：取前k个元素建立小根堆。堆顶就是目前第k个大的数。一次向后遍历原列表，对于列表中的元素，如果小于堆顶，则忽略该元素；如果大于堆顶，则堆顶替换为该元素，并且堆堆进行一次调整；遍历列表所有元素后，倒序弹出堆顶。 O(nlogk)

def sift(li, low, high):
	i = low
	j = 2 * i + 1
	temp = li[low]
	while j <= high:
		if j+1 <= high and li[j+1] > li[j]:
			j = j + 1
		if li[j] < temp:
			li[i] = li[j]
			i = j
			j = 2*i + 1
		else:
			break
		li[i] = temp


def topk(li, k):
	heap = li[:k]
	for i in range((k-1 -1)//2, -1, -1):
		sift(heap, i, k-1)
	# 小根堆建立完成
	for i in range(k, len(li)-1):
		if li[i] > heap[0]:
			heap[0] = li[i]
			sift(heap, i, k-1)
	# 遍历所有剩下元素
	for i in range(k-1, -1, -1):
		heap[0], heap[i] = heap[i], heap[0]
		sift(li, 0, i-1)
	
	return heap