前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >模拟堆(Java版)

模拟堆(Java版)

作者头像
用户10604450
发布2024-03-15 14:32:13
960
发布2024-03-15 14:32:13
举报
文章被收录于专栏:练习两年半

题目描述:

维护一个集合,初始时集合为空,支持如下几种操作:

  1. I x,插入一个数 x;
  2. PM,输出当前集合中的最小值;
  3. DM,删除当前集合中的最小值(数据保证此时的最小值唯一);
  4. D k,删除第 k 个插入的数;
  5. C k x,修改第 k 个插入的数,将其变为 x;

现在要进行 N 次操作,对于所有第 2 个操作,输出当前集合的最小值。

样例:

输入: 8 I -10 PM I -10 D 1 C 2 8 I 6 PM DM 输出: -10 6

如何手写一个堆?

堆的定义:根节点的值 小于等于 左右子节点的值(小根堆)。

存储采用一维数组(模拟最小堆,下标从1开始):x点的左儿子是:2x,x的右儿子是:2x+1

维护两个操作down 和 up

插入一个数

heap[ ++ size] = x; up(size)

求集合当中的最小值

heap[1]

删除最小值

heap[1] = heap[size]; size --; down(1);

为啥用最后一个元素覆盖第一个元素,因为删除第一个点不方便

删除任意一元素

heap[k] = heap[size]; size--; down(k); up(k)

后面两个操作只会执行一个或者不执行,因为变小才会向上走,变小向上走,或者不变

修改任意一个元素

heap[k] = x; down(x); up(x)

思路分析:

h[]: 存储堆里的元素

ph[]: 代表位置到堆的映射

hp[]: 代表堆到位置的映射

需要一个堆的数组是毋庸置疑的,创建下面两个数组的目的是什么呢?

ph[]数组

当执行删除第k个元素时,堆内元素会根据小根堆的性质不断移动,所以需要一个数组辅助去记住第几个插入的下标。 ph[k] = i:表示第k个插入的数在堆里面的下标为i。

hp[]数组

当我们交换了在堆里面下标为a、b这两个元素后,此时ph数组记录的插入顺序就错了,因为堆里面元素互换了。要想让ph数组正确,就需要互换ph[a],ph[b]。而要互换ph数组,就必须知道ph数组中的哪两个记录a、b的插入顺序。这时候你们可能会说,ph数组不是已经记录了吗?没错,ph数组是记录了,但是它是单向的,是ph数组指向堆元素下标的,而我们只知道堆元素的下标,我们怎么可能知道ph数组中的哪两个指向的a、b呢?所以必须开一个数组来存储堆里面下标为的元素是第几个插入的。使它们是双向的,这样才能互换ph数组。hp数组就是为了互换ph数组而服务的。

hp[j]:代表在堆的下标为j的元素 是第哪一次插入的

映射关系: 若hp[j] = k,则ph[k] = j。

详细代码(带注释)

代码语言:javascript
复制
import java.io.*;

public class Main {
    static int N=100010;
    static int []h=new int[N];
    static int []ph=new int[N],hp=new int[N];
    static int size=0;  //堆中元素的数量
    public static void main(String[] args) throws IOException {
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        int n=Integer.parseInt(br.readLine());
        int m=0;
        while (n-->0){
            String []s=br.readLine().split(" ");
            if("I".equals(s[0])){
                int x=Integer.parseInt(s[1]);
                h[++size]=x;
                m++;  //第几次插入
                ph[m]=size;
                hp[size]=m;
                up(size);
            }else if("PM".equals(s[0])){
                System.out.println(h[1]);
            }else if("DM".equals(s[0])){
                // 为啥用最后一个元素覆盖第一个元素?因为删除第一个点不方便
                heapSwap(1,size);
                // h[1] = h[size --]; 不能写成这种,因为要维持映射
                size--;
                down(1);
            }else if("D".equals(s[0])){
                int k=Integer.parseInt(s[1]);
                // 将第k次插入的元素,转换为堆中的下标
                int u=ph[k];
                heapSwap(u,size);
                size--;
                // 下面两个操作只会执行其中一个
                down(u); up(u);
            }else {
                int k=Integer.parseInt(s[1]);
                int x=Integer.parseInt(s[2]);
                h[ph[k]]=x;
                down(ph[k]); up(ph[k]);
            }
        }
        br.close();
    }

    // 应用于堆的交换
    private static void heapSwap(int i, int j) {
        swap(h,i,j); // 交换堆中两个元素的值
        swap(hp,i,j); // 维护堆到位置的映射
        swap(ph,hp[i],hp[j]);// 维护位置到堆的映射
    }

    //交换数组的两个元素
    private static void swap(int[] a, int i, int j) {
        int tmp=a[i];
        a[i]=a[j];
        a[j]=tmp;
    }

    // down操作是看一个元素是否需要往下移动
    private static void down(int i) {
      int t=i;
      // 如果左儿子存在 并且 左儿子比根节点小,就使 t 保存 左儿子下标
      if(i*2<=size &&h[i*2]<h[t]) t=2*i;
      if(i*2+1<=size &&h[i*2+1]<h[t]) t=2*i+1;
      if(t!=i){
          heapSwap(i,t);
          down(t);
      }
    }

    private static void up(int i) {
        // 只要根节点存在,并且u这个节点的值 比 根节点小,就需要交换
      if(i/2>0 && h[i/2]>h[i]){
         heapSwap(i,i/2);
         up(i/2);
      }
    }
}

难点:弄清三个数组分别是干什么的,并搞清之间的映射关系

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2024-03-15,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
对象存储
对象存储(Cloud Object Storage,COS)是由腾讯云推出的无目录层次结构、无数据格式限制,可容纳海量数据且支持 HTTP/HTTPS 协议访问的分布式存储服务。腾讯云 COS 的存储桶空间无容量上限,无需分区管理,适用于 CDN 数据分发、数据万象处理或大数据计算与分析的数据湖等多种场景。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档