5. 第k大元素

思路

在数组中找到第k大的元素
注意事项
    你可以交换数组中的元素的位置
样例
    给出数组 [9,3,2,4,8]，第三大的元素是 4
    给出数组 [1,2,3,4,5]，第一大的元素是 5，第二大的元素是 4，第三大的元素是 3，以此类推
分析

Python

class Solution:
    # @param k & A a integer and an array
    # @return ans a integer
    def kthLargestElement(self, k, A):
        A.sort()
        return A[-k]


if __name__ == "__main__":
    s = Solution()
    print(s.kthLargestElement(3, [9, 3, 2, 4, 8]))
    print(s.kthLargestElement(2, [1, 2, 3, 4, 5]))

Go

package main

import (
    "fmt"
    "sort"
)

/**
 * @param n: An integer
 * @param nums: An array
 * @return: the Kth largest element
 */
func kthLargestElement(n int, nums []int) int {
    // write your code here
    sort.Ints(nums)
    return nums[len(nums)-n]
}

func main() {
    l := []int{9, 3, 2, 4, 8}
    fmt.Println(kthLargestElement(3, l))
}