寻找数组中第K大数---使用堆

笔试时候遇上一个类似的问题，回来就搜了一下思路，

http://www.binghe.org/2011/05/find-kth-largest-number-in-disorder-array/

根据这里面的一个思路做的，

/***************************************************************
* 寻找数组中第k大的数， 即寻找最大的 k 个数中最小的那个，建立一
* 个 k 个元素的小根堆，堆顶即为 k 个数中最小的，然后依次读取数组
* 中的数字，比对，替换，调整堆，复杂度 n*log n ，不需要把数据一
* 次读入内存，且不会搞乱数据的顺序，数据很多时候有优势，但如果
* k 极大，就需要建多个堆。数据较少时不如快排的变形算法 
**************************************************************/ 

#include "sort_template.h" 

using namespace std;
void Build_heap(int*, int);
void swap(int &a, int &b);
void Show(int *, int);
int find_k(int*, int, int);
int main()
{
int arr[20];
srand(time(NULL));
for(int i = 0; i < 20; ++i)
arr[i] = rand()%50;
Quick(arr, 0, 20);
Show(arr, 20);
int i = find_k(arr, 9, 20);
cout << i << endl;
system("pause");
}
void swap(int &a, int &b)
{
a = a + b;
b = a - b;
a = a - b;
}
void Show(int *arr, int len)
{
for(int i = 0; i < len; ++i)
cout << arr[i] << "\t";
cout << endl;
}
void Build_heap(int *arr, int len)
{
for(int i = len/2 - 1; i >= 0; --i)
{
int index = i;
cout << i;
while(2*index+1 < len)
{
int min, lc = 2*index+1, rc = 2*index+2;
min = (lc < len && arr[lc] < arr[index]) ? lc : index;
min = (rc < len && arr[rc] < arr[min]) ? rc : min;
if(min != index)
{
swap(arr[min], arr[index]);
index = min;
}
else
break;
}
}
}
int find_k(int *arr, int k, int len)
{
Build_heap(arr, k);
for(int i = k; i < len; ++i)
{
if(arr[i] > arr[0])
swap(arr[i], arr[0]);
Build_heap(arr, k);
}
return arr[0];
}

有一个问题就是输出的时候，格式特别奇怪，再想想~~~~