题目
215 . Kth Largest Element in an Array
Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.
For example,
Given [3,2,1,5,6,4] and k = 2, return 5.
Note:
You may assume k is always valid, 1 ≤ k ≤ array’s length.
思路与代码
这道题目是寻找数组中第k大的数,首先想到的方法就是排序。最简单的方法就是进行采用冒泡排序的方法,循环k次就能找到第k大的数。直接上代码:
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
int len=nums.size();
#冒泡k次
for(int i=0;i<k;i++){
for(int j=0;j<len-i-1;j++)
{
if(nums[j]>nums[j+1])
{
swap(nums[j],nums[j+1]);
}
}
}
return nums[len-k];
}
};
leetcode中已经包含STL,可以直接用<alogrithm>
里面的函数.这种最简单的方法复杂度O(kn),k比较大时复杂度很高,一种办法是可以判断k与n/2的大小,来决定是升序还是降序。
自然会想到用其他更快的排序来搞定,比如O(nlgn)的快排,每次判断partition的位置的关系来缩小排序范围就可以了。分治思想。
int findKthLargest(vector<int>& nums, int k) {
int left = 0, right = nums.size() - 1;
while (true) {
int pos = partition(nums, left, right);
if (pos == k - 1) return nums[pos];
if (pos > k - 1) right = pos - 1;
else left = pos + 1;
}
}
int partition(vector<int>& nums, int left, int right) {
int pivot = nums[left];
int l = left + 1, r = right;
while (l <= r) {
if (nums[l] < pivot && nums[r] > pivot)
swap(nums[l++], nums[r--]);
if (nums[l] >= pivot) l++;
if (nums[r] <= pivot) r--;
}
swap(nums[left], nums[r]);
return r;
}
下面列出各种排序的复杂度
快排比较好理解,写起来也比较容易,但复杂度可能会陷入最坏情况,利用堆排序效果会更好。