題目

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

For example,
Given [3,2,1,5,6,4] and k = 2, return 5.

思路

利用快速排序的思想，用數組頭將數組兩分，左邊比它小，右邊比它大。如果這個元素正好是第k個，直接返回；如果比k大，則從比它小的那些數中找出第k大的；否則，從比它大的數中找出相應的數字。

代碼

class Solution(object):

    def quick_sort(self, nums, k):
        i = 0
        for j in range(1,len(nums)):
            if nums[j] > nums[0]:
                i += 1
                nums[i],nums[j] = nums[j],nums[i]
        nums[i],nums[0] = nums[0],nums[i]
        if i == k-1:
            return nums[i]
        elif i > k-1:
            return self.quick_sort(nums[:i], k)
        else:
            return self.quick_sort(nums[i+1:], k-i-1)

    def findKthLargest(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        if nums==None or len(nums)==0 or k<1:
            return -1
        return self.quick_sort(nums, k)