（2012-1）1000名求前30%

问题描述：

排序问题：1000个成绩输出前30%。

我这里输入十组数据，输出前3个

思路：

采用堆排序最佳。

方法一：

思路：采用堆排序，对1000个数据建大顶堆，输出堆顶最大值，调整300次，依次输出堆顶元素，这就是前300名；复杂度可以达到 nlogn，但是本题中似乎是 klogn

之后又看到 topK 问题的解法， 时间复杂度为 nlogk， 不知道是否优化

#include <cstdio>
#include <algorithm>
using namespace std;

//给定十个元素输出最大的前三个，元素从下标1开始存储
int heap[11] = {0, 3, 1, 2, 6, 5, 4, 9, 8, 7, 10};

int n = 10;      //元素个数

//对heqap数组在loe和high范围内进行调整
//low为父节点，high一般为堆的最后一个元素下标
void downAdjust(int low, int high){
    int i = low, j = i * 2;     //i为调整节点，j为左孩子
    while(j <= high){           //如果存在左孩子
        if(j + 1 <= high && heap[j + 1] > heap[j]){     //如果右孩子存在，且值大于左孩子
            j = j + 1;          //j来存储右孩子下标
        }
        //孩子中最大的权值大于要调整的节点
        if(heap[j] > heap[i]){
            swap(heap[j], heap[i]);
            i = j;          //继续调整以j为父节点的子树
            j = i * 2;
        }
        else
            break;
    }
}

//建树
void createHeap(){
    for(int i = n / 2; i >= 1; i--){        //i从第一个非叶节点开始，直到根结点
        downAdjust(i, n);
    }
}

void heapSort(){
    createHeap();   //建堆
    printf("%d", heap[1]);    //输出第一个最大值
    for(int i = n; i > 8; i--){     //倒着枚举，直到输出符合题目要求的数量
        swap(heap[i], heap[1]);     //交换heap[i]与堆顶
        downAdjust(1, i - 1);       //调整堆顶
        printf(" %d", heap[1]);     //调整过后，堆顶元素为最大值，输出
    }
    printf("\n");
}

int main(){
    heapSort();
    return 0;
}

方法二：topK 解法

思路：对前 300 个元素建小顶堆，此时堆顶为300个元素中的最小值，依次将后面的700个元素与堆顶元素比较，若大于堆顶元素，与其交换，调整小顶堆，直到700个元素比较完毕。此时，小顶堆中剩下的元素为前 300 个分数最大的

#include <cstdio>
#include <algorithm>
using namespace std;

//给定十个元素输出最大的前三个，元素从下标1开始存储
int heap[11] = {0, 3, 1, 2, 6, 5, 4, 9, 8, 7, 10};

int n = 10;      //元素个数
int k = 3;       //选出3个元素

//对heqap数组在loe和high范围内进行调整
//low为父节点，high一般为堆的最后一个元素下标
void downAdjust(int low, int high){
    int i = low, j = i * 2;     //i为调整节点，j为左孩子
    while(j <= high){           //如果存在左孩子
        if(j + 1 <= high && heap[j + 1] < heap[j]){     //如果右孩子存在，且值小于于左孩子
            j = j + 1;          //j来存储右孩子下标
        }
        //孩子中最大的权值小于于要调整的节点
        if(heap[j] < heap[i]){
            swap(heap[j], heap[i]);
            i = j;          //继续调整以j为父节点的子树
            j = i * 2;
        }
        else
            break;
    }
}

//建树
void createHeap(){
    for(int i = k / 2; i >= 1; i--){        //对前 k 个元素建树
        downAdjust(i, n);
    }
}

void heapSort(){
    createHeap();   //建堆
    int count = 1;
    while(count <= n - k){
        if(heap[1] < heap[k + count]){        //后面的700个元素与堆顶元素比较
            swap(heap[1], heap[k + count]);
            downAdjust(1, k);
        }
        count++;
    }
    for(int i = 1; i <= k; i++){
        printf("%d ", heap[i]);
    }
}

int main(){
    heapSort();
    return 0;
}