寻找数组中第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;
}