LeetCode 215
Kth Largest Element in an Array
- Problem Description:
返回数组的第k大元素,可采用快速排序的方法。快速排序每遍历一次就确定一个元素在数组中的最终位置,比该元素大的位于该元素右侧,比该元素小的位于该元素左侧。通过每次遍历后比较确定的位置和k 值大小,缩小区间范围,最终即可求出第k大元素。
具体的题目信息:
https://leetcode.com/problems/kth-largest-element-in-an-array/description/
- Example:
排序LeetCode 215 Kth Largest Element in an ArrayLeetCode 215 - Solution:
class Solution {
public:
int Partition(vector<int>& nums, int low, int high) {
int pivot = nums[low];
while(low<high) {
while(low<high && nums[high]>= pivot) high--;
nums[low] = nums[high];
while(low<high && nums[low]<= pivot) low++;
nums[high] = nums[low];
}
nums[low] = pivot;
return low;
}
int findKthLargest(vector<int>& nums, int k) {
if (nums.size() == ) return ;
if (nums.size() == ) return nums[];
k = nums.size()-k;
int low = , high = nums.size()-;
while(low<high) {
int pivot = Partition(nums, low, high);
if (pivot == k) {
break;
} else if (pivot<k) {
low = pivot+;
} else {
high = pivot-;
}
}
return nums[k];
}
};