标签:
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.
思路:
依据时间和空间复杂度,有多种方法。这里给出易于实现,基本能够满足面试、笔试要求的类似于快速排序的(递归)实现。
class Solution { public: int findKthLargest(vector<int>& nums, int k) { vector<int> leftvec; vector<int> rightvec; //以nums[0]作为分界元素,从nums[1]开始遍历 //比nums[0]小的元素放入leftvec,反之放入rightvec for(int i = 1; i < nums.size(); i++) { if(nums[i] <= nums[0]) leftvec.push_back(nums[i]); else rightvec.push_back(nums[i]); } //rightvec的长度即为大于nums[0]的元素的个数 int len = rightvec.size(); if(len >= k)//前k大的元素都在rightvec中,递归求解rightvec { return findKthLargest(rightvec, k); } else if(len == k - 1)//比nums[0]大的元素有k-1个,那么nums[0]即为所求结果 { return nums[0]; } else//比nums[0]大的元素有[0, k-2]个,说明第K大的元素在leftvec中,递归求解leftvec { return findKthLargest(leftvec, k - len - 1); } } };
【LeetCode 215】Kth Largest Element in an Array
标签:
原文地址:http://www.cnblogs.com/tjuloading/p/4580615.html