寻找数组中的第k个数

寻找数组中第k个最大的数跟快速排序的原理是类似的，快速排序的一次排序中是将数组的[i,j]段分为三部分[i,pivot-1],pivot,[pivot+1,j],那么pivot就是[i,j]中的第pivot-i+1个元素。那么把数组进行一次排序后，如果pivot就是第k个元素，则就达到了目的；如果pivot比k大，则表示第k大的元素在[i,pivot-1]中，因此现在就需要在[i,pivot-1]中进行寻找，[pivot+1,j]就不用再做处理；反之如果pivot比k小，那么第k大的元素就肯定在[pivot+1,j]中，因此就需要在[pivot+1,j]中寻找，[i,pivot-1]就不用再做处理。

整个算法的关键在于不断的对问题进行划分，使问题的规模越来越小

/*寻找数组中第i大的数*/

#include <stdio.h>
#include <time.h>
#include <stdlib.h>

void swap(int *a, int *b)
{
	int temp;

	temp = *a;
	*a = *b;
	*b = temp;
}

int PartitionTwo(int *arr, int low, int high)
{
	int sentinel = arr[high];
	int i,j = low - 1;				//j指向部分排序后数组中小于sentinel的最后一个节点

	for (i = low; i < high; i++)
	{
		if (arr[i] < sentinel)
		{
			j++;
			swap(&arr[i], &arr[j]);
		}
	}
	swap(&arr[j + 1], &arr[high]);

	return (j + 1);
}

int RandomPartition(int *arr, int low, int high)
{
	int randomNum = (int)((rand() % (high - low + 1)) + low);
	swap(&arr[randomNum], &arr[high]);
	return PartitionTwo(arr, low, high);
}

int find_kth_max(int *arr, int low, int high, int k)
{
	int partition, i;

	if (low == high)
		return arr[low];

	partition = RandomPartition(arr, low, high);
	i = partition - low + 1;

	if (i == k)
		return arr[partition];
	else if (i < k)
	{
		return find_kth_max(arr, partition + 1, high, k - i);
	}
	else
	{
		return find_kth_max(arr, low, partition - 1, k);
	}
}

int main()
{
	int arr[10010];
	int i, N;
	while (EOF != scanf("%d", &N))
	{
		for (i = 0; i < N; ++i)
			scanf("%d", &arr[i]);
		printf("%d\n", find_kth_max(arr, 0, N - 1, 1 + rand() % N));
	}

	return 0;
}