C++最小堆

#include <iostream>

class MinHeap
{
private:
    int _max;
    int _count;
    int *_ptr;

public:
    explicit MinHeap(int number)
    {
        _max = number;
        _count = 0;
        _ptr = new int(_max);
    }
    ~MinHeap()
    {
        if (_ptr)
        {

            delete[] _ptr;
            _ptr = nullptr;
        }
    }

    void push(int number)
    {
        if (_count >= _max)
        {
            return;
        }
        _ptr[_count] = number;
        int j = _count;
        while (j > 0)
        {
            int i = (j - 1) / 2; // 计算父节点位置,调整后的数组是从根节点开始,从上到下,从左到右
            if (_ptr[i] <= _ptr[j])
                break;

            std::swap(_ptr[i], _ptr[j]); // 交换值
            j = i;                       // 当前操作节点
        }
        _count++;
    }
    void pop()
    {
        if (_count < 1)
        {
            return;
        }
        _count--;
        int i = 0, j = 0, last = _count - 1;
        _ptr[i] = _ptr[_count]; // 交换最后一个和根节点值
        while (j < last)
        {
            j = 2 * i + 1;
            // 左节点大于右节点
            if (j < last && _ptr[j] > _ptr[j + 1])
                j++;

            if (_ptr[i] < _ptr[j])
                break;

            std::swap(_ptr[i], _ptr[j]); // 交换
            i = j;
        }
    }
    int count() { return _count; }
    int *data() { return _ptr; }
};

int main(int argc, char const *argv[])
{
    MinHeap myheap(256);
    int sample[] = {1};
    for (int i = 0; i < sizeof(sample) / sizeof(int); i++)
    {
        myheap.push(sample[i]);
        for (int k = 0; k < myheap.count(); k++)
        {
            printf("%d ", myheap.data()[k]);
        }
        printf("\n-------------------------\n");
    }
    myheap.pop();
    for (int k = 0; k < myheap.count(); k++)
    {
        printf("%d ", myheap.data()[k]);
    }
    printf("\n-------------------------\n");
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值