数据结构算法——1074. 寻找EMB富豪

题目

思路

注意数据过大，用scanf进行输入

正常的堆排序
分两步进行循环：

形成大根堆

void build_heap(int size) 
    {
        for (int i = size / 2 - 1; i >= 0; i--) 
        { // 倒数第二排开始, 创建大顶堆，必须从下往上比较
            down(i, size);                 // 否则有的不符合大顶堆定义
        }
    }
 

最后一个节点和第一个节点进行互换

build_heap(size);
for(int i = size - 1; i >= size - m; i--)
{
    swap(0,i);
    down(0,i);
}

代码

#include<iostream>
using namespace std;

struct to_heap
{
    int* arr;
    int size;
    to_heap(int n)
    {   size = n; 
        arr = new int[n];
    }
    ~to_heap()
    {delete[] arr;}

    void print(int n = 0)
    {
        if(!n) n = size;
        for(int i = size - 1; i > size - n - 1; i--)
            cout << arr[i] << (i == 0 ? "" : " ");
    }
    void zheng()
    {
        for(int i = 0 ; i < size; i++)
            cout << arr[i] << " ";
        cout << endl;
    }
    void swap(int x, int y)
    {
        int temp  = arr[x];
        arr[x] = arr[y];
        arr[y] = temp;
    }

    void down(int i, int n)
    {
        int parent = i;
        int child = 2 * i + 1;
    
        while (child < n) 
        {
            if (child + 1 < n && arr[child] < arr[child + 1]) { // 判断子节点那个大，大的与父节点比较
                child++;
            }
            if (arr[parent] < arr[child]) { // 判断父节点是否小于子节点
                swap(parent, child);     // 交换父节点和子节点
                parent = child;                 // 子节点下标 赋给 父节点下标
            }
            child = child * 2 + 1; // 换行，比较下面的父节点和子节点
        }
    }
    void build_heap(int size) 
    {
        for (int i = size / 2 - 1; i >= 0; i--) 
        { // 倒数第二排开始, 创建大顶堆，必须从下往上比较
            down(i, size);                 // 否则有的不符合大顶堆定义
        }
    }

    void heap_sort(int m)
    {
        if(m > size) m = size;
        build_heap(size);
        for(int i = size - 1; i >= size - m; i--)
        {
            swap(0,i);
            down(0,i);
        }
        print(m);
    }
};



int main()
{
    int n,m;
    cin >> n >> m;
    to_heap h(n);
    for(int i = 0; i < n; i++)
        scanf("%d",&h.arr[i]);
    h.heap_sort(m); 
    return 0;
}