c++堆详解【附代码】

目录

引言

一、堆的概念

二、堆的实现方式

三、堆的常见操作

四、堆的实现代码

结论

参考资料

---

引言

在计算机科学中，堆（Heap）是一种常见的数据结构，用于动态分配内存和管理数据。堆具有高效的插入和删除操作，并广泛应用于各种算法和数据结构中。本文将详细介绍堆的概念、原理、实现方式以及在C++中的应用，并提供相关的代码示例。

一、堆的概念

堆是一种特殊的树状数据结构，它满足以下两个条件：

1. 堆是一个完全二叉树（Complete Binary Tree），即除了最后一层外，其他层的节点数都是满的，并且最后一层的节点都尽量靠左排列。
2. 对于每个节点，它的值必须大于等于（或小于等于）其子节点的值。这种性质被称为堆属性（Heap Property）。

根据堆属性的不同，我们可以将堆分为最大堆和最小堆。在最大堆中，父节点的值大于等于其子节点的值；而在最小堆中，父节点的值小于等于其子节点的值。

二、堆的实现方式

堆可以使用数组或链表来实现。在本文中，我们将重点介绍使用数组实现堆的方法。

使用数组实现堆时，我们可以通过简单的数学关系来计算节点之间的父子关系，如下：

1. 对于索引为 i 的节点，其左子节点的索引为 2i+1，右子节点的索引为 2i+2。
2. 对于索引为 i 的节点，其父节点的索引为 (i-1)/2。

三、堆的常见操作

1. 插入元素：将新元素添加到堆的末尾，并根据堆属性进行调整。这通常涉及到通过交换节点来移动新元素，直到满足堆属性为止。插入元素的时间复杂度为 O(log n)，其中 n 是堆中元素的个数。
2. 删除堆顶元素：将堆顶元素删除，并将最后一个元素放在堆顶，然后根据堆属性进行调整。这通常涉及到通过交换节点来移动元素，直到满足堆属性为止。删除堆顶元素的时间复杂度为 O(log n)。
3. 获取堆顶元素：直接返回堆顶的元素，时间复杂度为 O(1)。
4. 堆排序：通过不断删除堆顶元素并将其放入结果中来对堆进行排序。堆排序的时间复杂度为 O(n log n)，其中 n 是堆中元素的个数。

四、堆的实现代码

下面是使用C++实现堆的示例代码：

#include <iostream>
#include <vector>

class Heap {
private:
    std::vector<int> heap;

    void heapifyUp(int index) {
        if (index == 0) {
            return;
        }

        int parentIndex = (index - 1) / 2;
        if (heap[parentIndex] < heap[index]) {
            std::swap(heap[parentIndex], heap[index]);
            heapifyUp(parentIndex);
        }
    }

    void heapifyDown(int index) {
        int leftChildIndex = 2 * index + 1;
        int rightChildIndex = 2 * index + 2;
        int largestIndex = index;

        if (leftChildIndex < heap.size() && heap[leftChildIndex] > heap[largestIndex]) {
            largestIndex = leftChildIndex;
        }

        if (rightChildIndex < heap.size() && heap[rightChildIndex] > heap[largestIndex]) {
            largestIndex = rightChildIndex;
        }

        if (largestIndex != index) {
            std::swap(heap[index], heap[largestIndex]);
            heapifyDown(largestIndex);
        }
    }

public:
    void insert(int value) {
        heap.push_back(value);
        heapifyUp(heap.size() - 1);
    }

    void remove() {
        if (heap.empty()) {
            return;
        }

        std::swap(heap[0], heap[heap.size() - 1]);
        heap.pop_back();
        heapifyDown(0);
    }

    int top() {
        if (heap.empty()) {
            throw std::runtime_error("Heap is empty");
        }

        return heap[0];
    }
};

int main() {
    Heap maxHeap;

    maxHeap.insert(5);
    maxHeap.insert(10);
    maxHeap.insert(3);

    std::cout << "Top element: " << maxHeap.top() << std::endl;

    maxHeap.remove();

    std::cout << "Top element after removal: " << maxHeap.top() << std::endl;

    return 0;
}

以上代码将输出：

Top element: 10
Top element after removal: 5

结论

堆是一种重要的数据结构，它具有高效的插入和删除操作，并广泛应用于各种算法和数据结构中。最大堆和最小堆是堆的两种常见类型，它们可以使用数组实现。通过堆，我们可以轻松地获取最大或最小的元素，并且堆排序是一种高效的排序算法。掌握堆的概念、实现方式以及相关操作，在解决问题时将会事半功倍。

参考资料

• Cormen, T. H., Leiserson, C. E., Rivest, R. L., & Stein, C. "Introduction to Algorithms." MIT Press, 2009.
• Sedgewick, R., & Wayne, K. "Algorithms." Addison-Wesley Professional, 2011.