C++模拟堆

模板题目

图片来源Acwing

堆的基础知识

代码实现

#include<iostream>

using namespace std;

const int N = 1e5+10;

int h[N], index;

void down(int x)
{
    int min = x;
    if(2 * x <= index && h[min] > h[x * 2])
    {
        min = x * 2;
    }
    if(2 * x + 1 <= index && h[min] > h[2 * x + 1])
    {
        min = 2 * x + 1;
    }
    if(min != x)
    {
        swap(h[min], h[x]);
        down(min);
    }
    return ;
}

int main()
{
    int n, m;
    cin >> n >> m;
    for(int i = 1; i <= n; i ++ )
    {
        scanf("%d", &h[i]);
    }
    
    index = n;
    
    for(int i = n / 2; i > 0; i -- )//n/2处是有子节点的最小的父节点，能够遍历全数组。
    //从n/2处开始能够缩短一些时间，但实际上从n处开始也能够实现排序的目的。
    {
        down(i);
    }
    
    while(m -- )
    {
        printf("%d ", h[1]);
        h[1] = h[index];
        index -- ;
        down(1);
        
    }
    return 0;
}

进阶题目

图源Acwing

解题思路

对于前三个操作来说都十分好处理，而为了实现最后两个操作，需要引入新的数组来反映第k个插入的数和其地址下标的关系：

所引入的三个数组的基本关系如下图所示

代码实现

#include<iostream>

using namespace std;

const int N = 1e5 + 10;

int h[N], hp[N], ph[N];

int index, insert_num;

void heap_swap(int a, int b)
{
    swap(h[a], h[b]);
    swap(hp[ph[a]], hp[ph[b]]);
    swap(ph[a], ph[b]);
}

void down(int x)
{
    int min = x;
    if(x * 2 <= index && h[2 * x] < h[min])
    {
        min = 2 * x;
    }
    if(x * 2 + 1 <= index && h[2 * x + 1] < h[min])
    {
        min = 2 * x + 1;
    }
    if(min != x)
    {
        heap_swap(x, min);
        down(min);
    }
}

void up(int x)
{
    if(x / 2 != 0 && h[x / 2] > h[x])
    {
        heap_swap(x / 2, x);
        up(x / 2);
    }
}

void insert(int x)
{
    index ++ ;
    insert_num ++ ;
    h[index] = x;
    hp[insert_num] = index;
    ph[index] = insert_num;
    up(index);
}

int main()
{
    int n;
    cin >> n;
    while(n -- )
    {
        string act;
        cin >> act;
        if(act == "I")
        {
            int x;
            scanf("%d", &x);
            insert(x);
        }
        else if(act == "PM")
        {
            printf("%d\n", h[1]);
        }
        else if(act == "DM")
        {
            heap_swap(1, index);
            index -- ;
            down(1);
        }
        else if(act == "D")
        {
            int k;
            scanf("%d", &k);
            int u = hp[k];//这里先要用u记住hp[K]的值，因为后面的heap_swap会改变hp[k]的值
            heap_swap(u, index);
            index -- ;
            down(u);
            up(u);
        }
        else if(act == "C")
        {
            int k, x;
            scanf("%d%d", &k, &x);
            h[hp[k]] = x;
            down(hp[k]);
            up(hp[k]);
        }
    }
    return 0;
}