堆数据结构小结

堆

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];