堆相关笔记

参考笔记：柳婼大佬的总结https://www.liuchuo.net/archives/2277

一、二次总结理由

关于二次总结是否有必要，我觉得是有的，参考学习别人的知识，内化的过程是一次总结的过程。每次参考同一个人的思维轨迹，确实能培养思维。该篇笔记整理理由：在PAT Advanced level的Heap Path一题，参考了关于堆的正序遍历的镜像。于是参考一下堆相关笔记。

二、堆数据结构主要算法(大根堆为例)

• 创建堆

void createHeap(){
    for(int i=n/2;i>=1;i--)
        downAjust(i,n);
}

• 向下调整：思想是，low代表孩子，high代表极限。我们在[low,high]范围进行调整，low和low+1为2*low的孩子。如果没有交换，那就break掉，交换了还要和上面父亲去比较。

void downAdjust(int low,int high){
    int i=low,j=i*2;//i为要调整的节点，j为左孩子
    while(j<=high){
        if(j+1<=high && heap[j+1]>heap[j]) j=j+1;
        if(heap[j]>heap[i]){
            swap(heap[j],heap[i]);
            i=j;j=i*2;
        }else break;
    }
}

• 删除堆顶元素

void deleteTop(){
    heap[1]=heap[n--];//用第n个数进行覆盖
    downAdjust(1,n);//之后进行向下调整第一个数
}

• 增加一个元素

void insert(int x){
    heap[++n]=x;
    upAdjust(1,n);
}

• 向上调整

void upAdjust(int low,int high){
    int i=high,j=i/2;
    while(j>=low){
        if(heap[j]<heap[i]){
            swap(heap[j],heap[i]);
            i=j;j=i/2;
        }else break;
    }
}

• 堆排序

将第一个数和顶部元素互换，进行向下调整（1，i-1）的范围，直到i==2

void heapSort(){
    createHeap();
    for(int i=n;i>=2;i--){
        swap(heap[i],heap[1]);
        downAdjust(1,i-1);
    }
}

堆排序的时间复杂度：O(nlogn) 空间复杂度为 O(1) 稳定性为不稳定

三、手写heap.h头文件

#ifndef HEAP_H_INCLUDED
#define HEAP_H_INCLUDED

#define maxn 1000

using namespace std;//这里面有swap函数

int heap[maxn];

void upAdjust(int low,int high){
    int i=high,j=i/2;
    while(j>=low){
        if(heap[j]<heap[i]){
            swap(heap[j],heap[i]);
            i=j;j=i/2;
        }else break;
    }
}

void downAdjust(int low,int high){
    int i=low,j=i*2;//i为要调整的节点，j为左孩子
    while(j<=high){
        if(j+1<=high && heap[j+1]>heap[j]) j=j+1;
        if(heap[j]>heap[i]){
            swap(heap[j],heap[i]);
            i=j;j=i*2;
        }else break;
    }
}

void createHeap(int n){
    for(int i=n/2;i>=1;i--)
        downAdjust(i,n);
}


void deleteTop(int n){
    heap[1]=heap[n--];//用第n个数进行覆盖
    downAdjust(1,n);//之后进行向下调整第一个数
}

void insert(int x,int n){
    heap[++n]=x;
    upAdjust(1,n);
}


void heapSort(int n){
    createHeap(n);
    for(int i=n;i>=2;i--){
        swap(heap[i],heap[1]);
        downAdjust(1,i-1);
    }
}
#endif // HEAP_H_INCLUDED

四、测试文件

#include <iostream>
#include "heap.h"
using namespace std;
int main(){
    extern int heap[maxn];
    extern int n;
    scanf("%d",&n);
    /**我们的堆序列是从[1,n]的*/
    for(int i=1;i<=n;i++) scanf("%d",&heap[i]);
    createHeap();//生成大根堆
    insert(4);//插入堆顶元素
    deleteTop();//删除堆顶
    heapSort();//排序
    for(int i=1;i<=n;i++) printf("%d%s",heap[i],i==n?"\n":" ");
    system("pause");
    return 0;
}

原文地址：https://www.cnblogs.com/littlepage/p/11617612.html