数组模拟堆

down和up这两个基本操作就可以实现其它操作
下面的down和up的比较逻辑都是基于建立小根堆

/**
 * 大的点下沉下来
 * down需要考虑递归down被交换的点是因为被交换下来的点和两个孩子节点之间没有确定的大小关系
 * 不像是up操作时,将原来的父节点交换下来,保证一定比孩子节点小
 */
void down(int u)
{
    int t = u;
    if (u * 2 <= size && h[u * 2] < h[t]) t = u * 2;
    if (u * 2 + 1 <= size && h[u * 2 + 1] < h[t]) t = u * 2 + 1;
    if (u != t)
    {
        heap_swap(u, t);
        down(t);
    }
}

/**
 * 小的点提升上去
 * up的感觉就是在堆已经建好后,再修改某个点时如果比父节点小才会用到
 * 因为此时堆中数据已基本有序,也就是父节点一定小于孩子节点,所以当前节点如果<父节点,两者交换后,之前的父节点不需要执行down操作,因为它和它的孩子节点相比它就是最小的
 */
void up(int u)
{
    while (u / 2 && h[u] < h[u / 2])
    {
        heap_swap(u, u / 2);
        u >>= 1;
    }
}

// O(n)建堆
for (int i = n / 2; i; i -- ) down(i);
posted @   0x7F  阅读(70)  评论(0编辑  收藏  举报
编辑推荐:
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
点击右上角即可分享
微信分享提示