查找中位数的O(N)算法和Kmin算法

最新推荐文章于 2024-11-01 15:57:30 发布

原创

最新推荐文章于 2024-11-01 15:57:30 发布 · 1.1w 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #null

本文介绍了如何运用快速排序的partition操作，在线性时间内找到数组的中位数，并探讨了寻找k小元素（Kmin）的方法。

利用快速排序的partition操作来完成O(N)时间内的中位数的查找：

#include <iostream>
#include <cassert>
#include <algorithm>
#include <iterator>

using namespace std;

int array[] = {1, 2, 10, 8, 9, 7, 5};
const int size = sizeof array / sizeof *array;

int partition(int *array, int left, int right)
{
	if (array == NULL)
		return -1;

	int pos = right;
	right--;
	while (left <= right)
	{
		while (left < pos && array[left] <= array[pos])
			left++;

		while (right >= 0 && array[right] > array[pos])
			right--;

		if (left >= right)
			break;

		swap(array[left], array[right]);
	}
	swap(array[left], array[pos]);

	return left;
}

int getMidIndex(int *array, int size)
{
	if (array == NULL || size <= 0)
		return -1;

	int left = 0;
	int right = size - 1;
	int midPos = right >> 1;
	int index = -1;

	while (index != midPos)
	{
		index = partition(array, left, right);

		if (index < midPos)
		{
			left