快排模板(附求第k大的数)

#include<cstdio>
#define REP(i, a, b) for(int i = (a); i < (b); i++)
using namespace std;

const int MAXN = 112;
int a[MAXN], n;

void sort(int l, int r)
{
	if(l >= r) return;
	int i = l, j = r, key = a[l];
	
	while(i < j)
	{
		while(i < j && a[j] >= key) j--;  //注意先从j开始, 也就是先从右边开始 
		if(i < j) a[i] = a[j];
		
		while(i < j && a[i] <= key) i++;
		if(i < j) a[j] = a[i];
	}
	
	a[i] = key;
	sort(l, i - 1); 
	sort(i + 1, r);
}

int main()
{
	scanf("%d", &n);
	REP(i, 0, n) scanf("%d", &a[i]);
	sort(0, n - 1);   //这里是n-1, 我写的版本是左闭右闭 
	REP(i, 0, n) printf("%d ", a[i]);
	puts("");
	return 0;
}

 

快排可以求第k大的数(如有些题目要求中位数), 利用左区间的个数来判断递归左区间还是右区间

//快排求第k大的数 
#include<cstdio>
#define REP(i, a, b) for(int i = (a); i < (b); i++)
#define _for(i, a, b) for(int i = (a); i <= (b); i++)
using namespace std;
 
const int MAXN = 112;
int a[MAXN], n, k;
 
void sort(int l, int r)  
{
	if(l >= r) return;  
	int i = l, j = r, key = a[l];  
	
	while(i < j)
	{
		while(i < j && a[j] >= key) j--;
		if(i < j) a[i] = a[j];
		
		while(i < j && a[i] <= key) i++;
		if(i < j) a[j] = a[i];
	}
	
	a[i] = key;
	if(i < k) sort(i + 1, r);    //k - 1 为所求的位置, 画个图就知道了 
	else if(i > k) return sort(l, i - 1); 
	else return; //并没有整个数组排完, 找到了就结束 
}
 
int main()
{
	scanf("%d", &n);
	_for(i, 1, n) scanf("%d", &a[i]);
	scanf("%d", &k);
	sort(1, n);  
	printf("%d\n", a[k]);
	return 0;
}

 

 

 

posted @ 2018-04-23 19:03  Sugewud  阅读(384)  评论(0编辑  收藏  举报