其实之前写的 6-11 求自定类型元素序列的中位数 是求第k大数字的一个特例。当时是将某一数组全部按照由大到小排列后,才取出中位数输出。然后昨天看数据结构时候才发现可以不完全排序,只拍第0到k个之间的就可以了。实现思想和快排类似:选取一个任意数,将小于这个数的数字排在该数的右边,大于这个数的数字排在该数的左边,比较此时k与这个数所在的位置i的大小,如果i>k,那么去继续排左边的数字,如果i<k,那么去排右边的数字,直至迭代到i == k。其实实质就是做一半的快排。
#include <stdio.h>
#include <stdlib.h>
int findKthLargest(int list[], int len, int Kth);
int solveKth(int list[], int left, int right, int k);
void showarray(int a[], int len);
int main()
{
int N, i, k;
scanf("%d %d", &N, &k);
int A[N];
for ( i=0; i<N; i++ )
scanf("%d", &A[i]);
printf("%d\n", findKthLargest(A, N, k));
system("pause");
}
int findKthLargest(int list[], int len, int Kth)
{
return solveKth(list, 0, len-1, Kth);
}
int solveKth(int list[], int left, int right, int k)
{
int i, tmp, j;
i = left;
j = right;
tmp = list[i];
while(i < j)
{
while(list[j]<tmp && i<j)
{
j--;
}
if(list[j]>=tmp && i<j)
{
list[i] = list[j];
i++;
}
while(list[i]>tmp && i<j)
{
i++;
}
if(list[i]<=tmp && i<j)
{
list[j] = list[i];
j--;
}
}
list[i] = tmp;showarray(list,10);
if((i+1) == k)
{printf("result out, i:%d, tmp:%d\n\n", i, tmp);
return tmp;}
if((i+1)>k)
{printf("i:%d, left:%d, right:%d, k:%d \n\n", i, left, i-1, k);
return solveKth(list, left, i-1, k);}
if((i+1)<k)
{printf("i:%d, left:%d, right:%d, k:%d \n\n", i, i+1, right, k);
return solveKth(list, i+1, right, k);}
}
void showarray(int a[], int len)
{
int j;
for(j=0; j<len; j++)
{
printf("a[%d]:%d\t", j, a[j]);
}
printf("\n");
}