Leetcode 215. Kth Largest Element in an Array(第二周作业)

先贴原题：

题意为：给出一个无序数组，然后指定一个整数值K,我们需要找出第k大的元素。

使用书上给出的方法：随便选择一个元素（我选择第size/2的元素），然后遍历数组，比这个元素小的丢到左边，比这个元素大的丢到右边，与元素相等的也记录下来。形成三个数组，倘若k小于等于右边数组的元素个数，则直接接着右边数组里找即可。倘若k比右边数组个数大，且小于等于右边数组的元素个数加中间数组元素的个数，则直接返回该元素的值。倘若大于右边数组和中间数组的元素个数的和，则在左边数组里找

class Solution {
public:
		void findNum(vector<int> nums, int numsSize, int select) {
			left.clear();
			equal.clear();
			right.clear();
			for (int i = 0; i < numsSize; i++) {
				if (nums[i] < select) left.push_back(nums[i]);
				else if (nums[i] == select) equal.push_back(nums[i]);
				else right.push_back(nums[i]);
			}
	}

	int findKthLargest(vector<int> nums, int k) {
        //使用中间的元素作为标准
		int select = nums[nums.size()>>1];
		int result = 0;
		findNum(nums, nums.size(), select);
        //确定下一步在哪个数组里找
		if (k <= right.size()) result = findKthLargest(right, k);
		else if (k > right.size() && k <= equal.size() + right.size()) result = select;
		else result = findKthLargest(left, k - equal.size() - right.size());
		return result;
	}
private:
	vector<int> left, equal, right;
};

该算法的时间复杂度为O(nlogn);

空间复杂度为O(n),即申请了三个vector<int>，用于存放每次divide的元素。