码迷,mamicode.com
首页 > 其他好文 > 详细

POJ 2388 Who's in the Middle 快排解法

时间:2014-06-15 16:21:17      阅读:214      评论:0      收藏:0      [点我收藏+]

标签:class   blog   code   2014   art   算法   

又是一题快速排序的题目,活用快排求某个位置的数。

这次完善一下自己的基础,把快排代码规范化和增加一个random算法,进一步确保不会出现最坏情况。

思路和前一道题差不多,不过是求第k个数了,这里的第k个数是中序数。

花了点时候整理下代码,果然变得十分工整了。

#include <cstdio>
#include <stdlib.h>
#include <algorithm>
#include <time.h>
using namespace std;
const int SIZE = 10000;
int arr[SIZE];

int partition(int *arr, int low, int up)
{
	for (int j = low; j < up; j++)
		if (arr[j] < arr[up]) 
			swap(arr[low++], arr[j]);

	swap(arr[low], arr[up]);
	return low;
}

int selectK_1(int *arr, int low, int up, int k)
{
	int r = rand() % (up - low + 1) + low;
	swap(arr[r], arr[up]);

	int m = partition(arr, low, up);
	int lowerParts = m - low + 1;

	if (lowerParts == k) return arr[m];
	if (k < lowerParts) return selectK_1(arr, low, m-1, k);
	return selectK_1(arr, m+1, up, k - lowerParts);
}

int main()
{
	srand((int)time(NULL));
	int N;
	while (~scanf("%d", &N))
	{
		for (int i = 0; i < N; i++)
		{
			scanf("%d", &arr[i]);
		}
		printf("%d\n", selectK_1(arr, 0, N-1, (N>>1)+1));
	}
	return 0;
}


POJ 2388 Who's in the Middle 快排解法,布布扣,bubuko.com

POJ 2388 Who's in the Middle 快排解法

标签:class   blog   code   2014   art   算法   

原文地址:http://blog.csdn.net/kenden23/article/details/30289893

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!