Kth Largest Element in an Array
Given an integer array nums
and an integer k
, return the kth
largest element in the array.
Note that it is the kth
largest element in the sorted order, not the kth
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 <= 105
-104 <= nums[i] <= 104
Solutions
Approach - Brute Force Technique
public class Solution {
public int FindKthLargest(int[] nums, int k) {
Array.Sort(nums);
k=nums.Length-k;
return nums[k];
}
}
Complexity
Time Complexity: O(n log n)
Space Complexity: O(1)
Approach - Min Heap Technique
public class Solution {
public int FindKthLargest(int[] nums, int k) {
// Create a Min Heap
SortedSet<(int num, int index)> minHeap = new SortedSet<(int num, int index)>();
int i = 0;
// Add the first 'k' numbers in the min heap
for (; i < k; i++)
minHeap.Add((nums[i], i));
// Go through the remaining numbers of the array, if the number from the array is bigger than the
// top (smallest) number of the min-heap, remove the top number from heap and add the number from array
for (; i < nums.Length; i++) {
if (nums[i] > minHeap.Min.num) {
minHeap.Remove(minHeap.Min);
minHeap.Add((nums[i], i));
}
}
// The root of the heap has the 'Kth' largest number
return minHeap.Min.num;
}
}
Complexity
Time Complexity: O(n log k)
Space Complexity: O(k)
Last updated