# -*- coding:utf-8 -*-
# @Time    : 2019/8/15 19:30
# @Author  : Junwu Yu
'''
在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:

输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:

你可以假设 k 总是有效的，且 1 ≤ k ≤ 数组的长度。
'''

import heapq

class Solution:
    def findKthLargest(self, nums, k) -> int:
        return heapq.nlargest(k, nums)[-1]


s = Solution()
nums = [3, 2, 3, 1, 2, 4, 5, 5, 6]
print(s.findKthLargest(nums, 2))
