AcWing 839.模拟堆

Cyber_Wz

已于 2022-04-08 15:51:01 修改

阅读量425

点赞数

分类专栏： AcWing题解文章标签：蓝桥杯 c语言

于 2022-02-12 12:59:12 首次发布

本文链接：https://blog.csdn.net/m0_50564748/article/details/122893535

版权

AcWing题解专栏收录该内容

70 篇文章 13 订阅

订阅专栏

这篇博客介绍了如何使用额外的数组ph和hp记录堆中元素的插入顺序，并详细阐述了堆排序中的堆交换、上浮和下沉操作。通过示例展示了如何在已排序的堆中进行插入、查找第k个元素、删除最小元素以及更新元素值的操作。

摘要由CSDN通过智能技术生成

对于堆介绍可以看这篇博客 Acwing 838.堆排序_m0_50564748的博客-CSDN博客

这篇主要讲一下，如何找到第k个插入的数。

那么此时需要额外的两个数组来进行记录，分别是ph[k],hp[k].

ph[k]:第k个插入的数在堆中的编号！

hp[k]:在堆中编号为k的元素是第几个插入的！

这其实是一个映射关系！

现在我们想要交换堆中编号1，2的元素！

839. 模拟堆 - AcWing题库

#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
const int N=1e5+10;
int h[N],hp[N],ph[N],n,m,sizee;
//ph[k]插入的第k个数在堆里(树里)下标是几
//hp[k]堆里的编号为k的点是第几个插入的
void heap_swap(int a,int b)
{
    //第x次 插入的编号为 a ph[x]=a hp[a]=x h[a]
    //第y次 插入的编号为 b ph[y]=b hp[b]=y h[b]
    swap(ph[hp[a]],ph[hp[b]]);//交换之后，ph[x]=b,ph[y]=a;
    swap(hp[a],hp[b]);//交换之后，hp[a]=y,hp[b]=x;
    swap(h[a],h[b]);//交换元素值
}

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

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

int main()
{
    scanf("%d",&n);
    while(n--)
    {
        char op[10];
        scanf("%s",op);
        int k,x;
        if(!strcmp(op,"I")) 
        {
            scanf("%d",&x);
            sizee ++ ;
            m ++ ;
            ph[m] = sizee, hp[sizee] = m;
            //ph[m]=sizee 第m次插入的数在堆中编号为sizee
            //hp[sizee]=m 编号为sizee的数是第m次插入
            h[sizee]=x;
            up(sizee);
        }
        else if(!strcmp(op,"PM")) printf("%d\n",h[1]);
        else if(!strcmp(op,"DM"))
        {
            heap_swap(1,sizee),sizee--;
            down(1);
        }
        else if(!strcmp(op,"D"))
        {
            scanf("%d",&k);
            k=ph[k];
            heap_swap(k,sizee);
            sizee--;
            down(k),up(k);
        }
        else
        {
            scanf("%d%d",&k,&x);
            k=ph[k];
            h[k]=x;
            down(k),up(k);
        }
    }
    return 0;
}