数组中的第K个最大元素

未排序的数组，第K个最大元素，使用小根堆实现，小根堆的定义就是根节点是最小值

class Solution(object):
    def findKthLargest(self, nums, k):
    	def shift(i,k):
            while True:
                t = i
                if (i*2+1)<k and nums[i]>nums[2*i+1]: #左子节点存在，且小于根节点，则交换，使根节点为最小值  
                    t=2*i+1
                if (i*2+2)<k and nums[t]>nums[2*i+2]: 
                    t=2*i+2
                if t==i:
                    break
                else:
                    nums[i],nums[t]=nums[t],nums[i]
                    i=t
        #O(k):建立大小为K的最小堆
        for i in range(k,-1,-1):
            shift(i,k)
        #O((N-k)logK)，剩余元素依次比较替换
        for i in range(k,len(nums)):
            if nums[0]<nums[i]:
                nums[0]=nums[i]
                shift(0,k)
        return nums[0]
        #sum=O(Nlogk-k(logK-1))

class Solution:#一句代码解决，找nums数组中前k个最大元素
    def findKthLargest(self, nums, k):
        return heapq.nlargest(k, nums)[-1]