查找中位数的O(N)算法和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 = index + 1;
		}
		else if (index > midPos)
		{
			right = index - 1;
		} 
		else
		{
			break;
		}
	}

	assert(index == midPos);
	return array[index];
}

void main()
{
	int value = getMidIndex(array, size);

	cout << "value: " << value << endl;
}

寻找kmin

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

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

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

		if (index < k)
		{
			left = index + 1;
		}
		else if (index > k)
		{
			right = index - 1;
		} 
		else
		{
			break;
		}
	}

	assert(index == k);
	return array[index];
}

    原文作者:查找算法
    原文地址: https://blog.csdn.net/alexingcool/article/details/7805228
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞