feat(work): add solution for LeetCode 215 Kth Largest Element in an Array

This commit is contained in:
Prad Nukala
2026-09-14 12:21:55 -04:00
parent 393d72c9fe
commit 2066856d15
@@ -0,0 +1,50 @@
"""
215. Kth Largest Element in an Array
Difficulty: Medium
https://leetcode.com/problems/kth-largest-element-in-an-array/
──────────────────────────────────────────────────
Given an integer array nums and an integer k, return the k^th largest
element in the array.
Note that it is the k^th largest element in the sorted order, not the
k^th distinct element.
Can you solve it without sorting?
Example 1:
Input: nums = [3,2,1,5,6,4], k = 2
Output: 5
Example 2:
Input: nums = [3,2,3,1,2,4,5,5,6], k = 4
Output: 4
Constraints:
• 1 <= k <= nums.length <= 10^5
• -10^4 <= nums[i] <= 10^4
"""
import heapq
class Solution:
def findKthLargest(self, nums: List[int], k: int) -> int:
for i in range(len(nums)):
nums[i] = -nums[i]
heapq.heapify(nums)
for _ in range(k - 1):
heapq.heappop(nums)
return -heapq.heappop(nums)