《编程之美》——寻找最大的K个数(P139)
编程环境:Visual Studio 2010
解法一:寻找最大的K个数最容易想到的方法是排序然后找出最大的K个数,在这里快速排序和堆排序都是不错的选择,他们的平均时间复杂度都是O(N*logN)。
堆排序代码如下
#define LeftChild(i) (2*(i)+1)
static void swap(ElementType *lhs, ElementType *rhs)
{
ElementType tmp;
tmp = *lhs;
*lhs = *rhs;
*rhs = tmp;
}
static void percdown(int *arr, int i, int n)//大根堆
{
ElementType tmp;
int child;
for(tmp = arr[i]; LeftChild(i) < n; i = child)
{
child = LeftChild(i);
if(child != n-1 && arr[child + 1] > arr[child])
child++;
if(tmp < arr[child])
arr[i] = arr[child];
else
break;
}
arr[i] = tmp;
}
void heap_sort(ElementType *arr, int n)
{
int i;
for(i = n/2; i >= 0; i--)
percdown(arr, i, n);
for(i = n-1; i >