[AcWing]AcWing 839. 模拟堆（C++实现）堆排序模板题（带映射）

1. 题目

2. 读题（需要重点注意的东西）

思路：
本题和普通堆排序的最大的不同之处在于存在一种映射关系，来存储第k个插入的数在堆中的下标是多少

由于要删除第 k 个插入的数，因此需要ph数组和hp数组
ph[k]:保存第k个插入的数的下标cnt
hp[cnt]:保存堆中下标是cnt的点是第几个插入的

3. 解法

---------------------------------------------------解法---------------------------------------------------

#include <iostream>
#include <algorithm>
#include <string.h>

using namespace std;

const int N = 100010;

/* 
    由于要删除第 k 个插入的数，因此需要ph数组和hp数组
    ph[]:用p指向堆h
    hp[]:用h指向p
*/
int h[N], ph[N], hp[N], cnt;

void heap_swap(int a, int b)
{
    swap(ph[hp[a]],ph[hp[b]]); // ph相应的值需要交换
    swap(hp[a], hp[b]); // hp相应的值也需要交换
    swap(h[a], h[b]); // 交换h数组对应的值
}

void down(int u)
{
    int t = u;
    if (u * 2 <= cnt && h[u * 2] < h[t]) t = u * 2;
    if (u * 2 + 1 <= cnt && h[u * 2 + 1] < h[t]) t = u * 2 + 1;
    if (u != t)
    {
        heap_swap(u, t);
        down(t);
    }
}

void up(int u)
{
    while (u / 2 && h[u] < h[u / 2])
    {
        heap_swap(u, u / 2);
        u >>= 1;
    }
}

int main()
{
    int n, m = 0;
    scanf("%d", &n);
    while (n -- )
    {
        char op[5];
        int k, x;
        scanf("%s", op);
        /* 
            strcmp(str1,str2)，比较两个字符串
            头文件在#include<string.h>中
            1. 如果返回值小于 0，则表示 str1 小于 str2；
            2. 如果返回值大于 0，则表示 str1 大于 str2;
            3. 如果返回值等于 0，则表示 str1 等于 str2。
        */
        if (!strcmp(op, "I"))
        {
            scanf("%d", &x);
            cnt ++ ;
            m ++ ; // 用m来记录字符出现的序号
            ph[m] = cnt, hp[cnt] = m; // m与cnt互相映射
            h[cnt] = x;
            up(cnt);
        }
        else if (!strcmp(op, "PM")) printf("%d\n", h[1]);
        else if (!strcmp(op, "DM"))
        {
            heap_swap(1, cnt);
            cnt -- ;
            down(1);
        }
        else if (!strcmp(op, "D"))
        {
            scanf("%d", &k);
            k = ph[k];
            heap_swap(k, cnt);
            cnt -- ;
            up(k);
            down(k);
        }
        else
        {
            scanf("%d%d", &k, &x);
            k = ph[k];
            h[k] = x;
            up(k);
            down(k);
        }
    }

    return 0;
}

4. 可能有帮助的前置习题

• [AcWing]838. 堆排序（C++实现）堆排序模板题

5. 所用到的数据结构与算法思想

6. 总结

堆的相关运算，带映射关系，用数组实现的模板题，推荐完全背下来。
与普通堆排序最大的不同，即映射关系的构建和交换：

1. 构建：

        if (!strcmp(op, "I"))
        {
            scanf("%d", &x);
            cnt ++ ;
            m ++ ; // 用m来记录字符出现的序号
            ph[m] = cnt, hp[cnt] = m; // m与cnt互相映射
            h[cnt] = x;
            up(cnt);
        }

2. 交换
   堆排序进行down和up操作时，要同步交换ph和hp数组

void heap_swap(int a, int b)
{
    swap(ph[hp[a]],ph[hp[b]]); // ph相应的值需要交换
    swap(hp[a], hp[b]); // hp相应的值也需要交换
    swap(h[a], h[b]); // 交换h数组对应的值
}