堆数据结构小结

1.算法分析

down(x)函数:从x开始往下使用筛选法调整堆,up(x)函数:从x开始往上使用筛选法调整堆

堆的基本操作(以小根堆为例):

1.插入一个数: heap[++size] = x; up(size)

2.求集合当中的最小值:heap[1]

3.删除最小值:heap[1] = heap[size];size–;down(1)

4.删除任意一个元素:heap[k] = heap[size];size–;down(k) or up(k) (比原位置的数字大,就往下走;比原位置的数字小,就往上走)

5.修改任意一个元素:heap[k] = x; down(k) or up(k)(比原位置的数字大,就往下走;比原位置的数字小,就往上走)

2.模板

// 筛选法向下调整堆
// down(x):从堆内下标为x的地方开始向下调整堆
void down(int u) {
   
    int t = u;  // t为堆的父节点和两个子节点中值最小的那个
    if (u * 2 <= size && heap[u * 2] < heap[t]) t = u * 2;  // u * 2 <= size是为了保证数据不越界,即保证u只能访问到树的倒数第二层的最后一个结点
    if (u * 2 + 1 <= size && heap[u * 2 + 1] < heap[t]) t = u * 2 + 1;
    if (u != t) {
   // 把父节点和最小的子节点的值交换
        swap(heap[t], heap[u]);
        down(t);  // 交换后可能导致子节点t不构成堆,所以要从t开始继续往下调整
    }
}

// up操作
// up(x):从堆内下标为x的地方开始向上调整堆
void up(int u) {
   
    while (u / 2 && heap[u / 2] > heap[u]) {
     // 和它的父节点比较,如果父节点比自己大,那就交换父节点和子节点
        heap_swap(u, u / 2);
        u = u / 2;
    }
}

// 初始化建堆(时间复杂度O(n))
    for (int i = 1; i <= n; ++i) cin >> heap[i];
    size = n;
    for (int i = n / 2; i >= 0; --i) down(i);

// 插入操作(插入数字为num的结点)
    int num;
    scanf("%d", &num);
    size++;
    m++;  // m表示本次为第m次插入
    ph[m] = size, hp[size] = m;  // 记录输入顺序和下标顺序:ph[m] = size:第m次插入的堆内下标为size,
                                 //  hp[size] = m:堆内下标为size的结点是第m次插入的
    heap[size] = num;  // 往堆内放入元素
    up(size);  // 从size结点往上调整

// 输出堆顶
    printf("%d\n", heap[1]);

// 删除最小值
    heap_swap(1, size);  // 交换堆顶结点和size结点
    size--;
    down(1);  // 从1号结点开始往下调整

// 删除第k个输入的结点
    int k;
    scanf("%d", &k);
    k = ph[k];  // 找到第k个结点在堆内的下标位置
    heap_swap(k, size);  // 交换
    size--;  // 减少一个结点
    up(k);  // 分别向上和向下调整,但本质上只会进行一个操作,因为交换后只会只会破坏向上或者向下其中一个的结构
    down(k);

// 修改第k个输入的结点
    int k, x;  
    scanf("%d %d", &k, &x);
    k = ph[k];  
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值