[leetcode]215. Kth Largest Element in an Array

题意：

给一个数组，求出数组中的第K大数，比如[3,2,1,5,6,4] 求出第2大数，则为5

分析：

复习堆排序，对于这道题来讲，找第k大数，可以通过一个小顶堆实现。首先取数组中的前k个数组成小顶堆（具体堆的调整可以百度，或者看代码应该就理解），此时堆顶的数字就是这k个数字里面的第K大数，然后从每次从剩余的数组中拿出一个数字和堆顶的数字进行比较，如果比堆顶数字小，则直接丢弃；如果大于堆顶数字，则替换该数字，重新进行堆调整，调整完成之后堆顶的数字依旧是已经处理过的数字中的第K大数。如此重复执行，直到数据处理完毕，此时堆顶的数字就是第K大数

代码：

#!/usr/bin/env python
class Solution(object):
    def __init__(self):
        self.nums_t = []
    def min_heap(self, i, l):
        L = i * 2
        R = i * 2 + 1
        min = i
        if L <= l and self.nums_t[L-1] < self.nums_t[i-1]:
            min = L
        if R <= l and self.nums_t[R-1] < self.nums_t[min-1]:
            min = R
        if min != i:
            tp = self.nums_t[min-1]
            self.nums_t[min-1] = self.nums_t[i-1]
            self.nums_t[i-1] = tp
            self.min_heap(min, l)

    def build_min_heap(self):
        l = len(self.nums_t)
        for i in range(l/2, 0, -1):
            self.min_heap(i, l)

    def findKthLargest(self, nums, k):
        self.nums_t = nums[0:k]
        self.build_min_heap()
        nums_leave = nums[k:]
        LL = len(nums_leave)
        for i in range(0, LL):
            if nums_leave[i] > self.nums_t[0]:
                self.nums_t[0] = nums_leave[i]
                self.min_heap(1, len(self.nums_t))
        return self.nums_t[0]
# 下面为测试代码
if __name__ == '__main__':
    nums = [4,1,3,2,16,9,10,14,8,7]
    l = 5
    s = Solution()
    ans = s.findKthLargest(nums, l)
    print "ans : ", ans

如代码所示，对于前K个数字进行建小顶堆，然后对于剩余的数字进行比较处理，最后堆顶数字即为所求