查找前K大的数

题目描述如上。

思路：把前k大的都弄到数组最右边，然后对这最右边k个元素排序，再输出
关键 ：O(n)时间内实现把前k大的都弄到数组最右边

此处的PPT来源：

如果不小心冒犯了老师的版权，请通知我一下，我里面删除内容。。。

最后贴一下代码：

#include <iostream>
#include <algorithm>
#include <functional>

using namespace std;

void findKth(int a[], int s, int e, int k);

void swap(int &a, int &b);

const int maxn = 100005;
int a[maxn];

int main() {
	int n;
	cin >> n;
	for(int i = 0; i < n; i++) {
		cin >> a[i];
	}
	int k;
	cin >> k;
	findKth(a, 0, n - 1, k);
	sort(a + n - k, a + n);
	for(int i = n - 1; i >= n - k; i--) {
		cout << a[i] << endl;
	}
	return 0;
}

void findKth(int a[], int s, int e, int k) {
	if(s >= e) {
		return;
	}
	int t = a[s];
	int i = s;
	int j = e;
	while(i != j) {
		if(i < j && a[j] >= t) {
			j--;
		}
		swap(a[i], a[j]);
		if(i < j && a[i] <= t) {
			i++;
		}
		swap(a[i], a[j]);
	}
	int right = e + 1 - i;
	if(right == k) {
		return;
	} else if(right > k) {
		findKth(a, i + 1, e, k);
	} else {
		findKth(a, s, i - 1, k - right);
	}
}

void swap(int &a, int &b) {
	int temp = a;
	a = b;
	b = temp;
}