中位数的查询（C++）

中位数的查询

题目要求

在一组长度为n的无序的数组中，找出第k小的元素（0<k<n）

分而治之的策略

与快速排序类似，我们也将函数进行划分。
首先我们初始化一个特殊值，不妨设为数组的第一个元素。
以这个特殊值为主体进行划分，将小于这个特殊值的元素放到这个特殊值的左侧，将大于这个特殊值的元素放到这个特殊值的右侧，进行这一次划分后，这个特殊值的元素就固定了，它左侧的值全为小于它的，它右侧的值全为大于它的。
那么我们就好办了，只需要将这个元素所在位置与k进行对比，如果k相对大，那么我们只需要在这个元素右边再进行寻找即可。
如果k相对小那么我们只需要在这个元素左边进行寻找即可。
直到找到为止。

举例

第一步：初始化，不妨设k值为8。

第二步：找到tmp(56)的位置，保证tmp值左侧均为小于tmp的值，右侧均为大于tmp的值。最终找到tmp的位置为5，k大于5，所以要在位置5的右侧找。

第三步：找到tmp(89)的位置，最终找到tmp的位置为9，相对位置为4，k小于4，所以要在位置9的左侧寻找。

第四步：找到tmp(89)的位置，最终找到tmp的位置为8，相对位置为3，k小于3，所以要在位置8的左侧寻找。

第五步：找到tmp(78)的位置，最终找到tmp的位置为7,找到目标位置。

代码实现

判断是否符合题意函数

int SelectKMin(int* arr, int n, int k)
{
	if (arr == nullptr || n < 1 || k<1 || k>n) return -1;
	return SelectK(arr, 0, n - 1, k);
}

寻找k位置函数

int SelectK(int* arr, int left, int right, int k)
{
	if (left == right && k == 1) return arr[left];
	int mid = Partition(arr, left, right);
	int tmp = mid - left + 1;//记录相对位置
	if (tmp == k)
	{
		return arr[mid];
	}
	if (tmp > k)
	{
		return SelectK(arr, left, mid-1, k);
	}
	if (tmp < k)
	{
		k = k - tmp;
		return SelectK(arr, mid + 1, right, k);
	}
	return -1;
}

划分函数

int Partition(int* arr, int left, int right)
{
	int i = left;
	int j = right;
	int tmp = arr[left];
	while (i < j)
	{
		while (i<j && arr[j]>tmp) j--;
		arr[i] = arr[j];
		while (i < j && arr[i] <= tmp) i++;
		arr[j] = arr[i];
	}
	arr[i] = tmp;
	return i;
}

总体代码

思路与快速排序类似

#include<iostream>
using namespace std;

int Partition(int* arr, int left, int right)
{
	int i = left;
	int j = right;
	int tmp = arr[left];
	while (i < j)
	{
		while (i<j && arr[j]>tmp) j--;
		arr[i] = arr[j];
		while (i < j && arr[i] <= tmp) i++;
		arr[j] = arr[i];
	}
	arr[i] = tmp;
	return i;
}

int SelectK(int* arr, int left, int right, int k)
{
	if (left == right && k == 1) return arr[left];
	int mid = Partition(arr, left, right);
	int tmp = mid - left + 1;//记录相对位置
	if (tmp == k)
	{
		return arr[mid];
	}
	if (tmp > k)
	{
		return SelectK(arr, left, mid-1, k);
	}
	if (tmp < k)
	{
		k = k - tmp;
		return SelectK(arr, mid + 1, right, k);
	}
	return -1;
}

int SelectKMin(int* arr, int n, int k)
{
	if (arr == nullptr || n < 1 || k<1 || k>n) return -1;
	return SelectK(arr, 0, n - 1, k);
}

int main()
{
	int arr[] = { 56,23,78,45,90,89,12,34,56,67,89,100 };
	int n = sizeof(arr) / sizeof(arr[0]);
	int k = 0;
	
	for (int i = 1; i <= n; i++)
	{
		k = i;
		int kmin = SelectKMin(arr, n, k);
		cout << kmin << endl;
	}
	return 0;
}