第k大元素

题目描述：在数组中找到第k大的元素

样例：给出数组 [9,3,2,4,8]，第三大的元素是 4；给出数组 [1,2,3,4,5]，第一大的元素是 5，第二大的元素是 4，第三大的元素是 3，以此类推

和上一道中位数的题（详见：点击打开链接）是一模一样的，需要注意的是，数组既然要按升序排列，那么第k大元素就是数组的倒数第k位，也就是索引为n - k的元素，还是按照快排的思想，通过一个分割函数找到每次枢轴应该排在的位置，比较这个位置和n - k的前后，如果恰好是n - k，直接返回；如果小于n
- k，则说明需要在枢轴所在后面的数组中找寻；如果大于n - k，则说明需要在枢轴所在前面的数组中找寻。这个思路如果还有疑问，请移步我刚才给的链接。

那么代码几乎跟“中位数”的题是一样的，如下：

class Solution:
    # @param k & A a integer and an array
    # @return ans a integer
    def kthLargestElement(self, k, A):
        n = len(A)
        if n == 0:
            return None
        begin, end = 0, n - 1
        index = self.partition(A, 0, n - 1)
        while index != n - k:
            if index > n - k:
                end = index - 1
                index = self.partition(A, begin, end)
            else:
                begin = index + 1
                index = self.partition(A, begin, end)

        return A[index]

    def partition(self, A, begin, end):
        if begin >= end:
            return begin
        pivot = A[begin]
        index = begin
        for i in range(begin + 1, end + 1):
            if A[i] <= pivot:
                index += 1
                A[i], A[index] = A[index], A[i]
        A[begin], A[index] = A[index], A[begin]
        return index

这两道题，包括前面排列颜色的问题，都是快排的应用，所以，有时候学习算法，确实需要理解它背后的原理，而不仅仅是会写出这一个问题的代码。