堆排序简单实现

来源:互联网 发布:知峰竹纤维厂家 编辑:程序博客网 时间:2024/06/03 07:11

(算法导论排序算法部分第一个就介绍堆排序,也得复习复习。)

Wiki(http://zh.wikipedia.org/wiki/%E5%A0%86%E6%8E%92%E5%BA%8F#C.E8.AF.AD.E8.A8.80)上也有介绍及代码实现,基本上都是一样的。

摘录堆排序定义如下:

堆排序(Heapsort)是指利用堆这种数据结构所设计的一种排序算法。堆是一个近似完全二叉树的结构,并同时满足堆性质:即子结点的键值或索引总是小于(或者大于)它的父节点。

然后实现代码如下:

#include <iostream>#include <cmath>#include <algorithm>using namespace std;#define ARRAY_SIZE 1000#define LEFT_CHILD(i) (2 * i)#define RIGHT_CHILD(i) (2 * i + 1)int data[ARRAY_SIZE];//前提root的左右子树都是最大堆//用来平衡最大二叉堆//找出最大子树根,然后替换,递归void Max_Heapify(int data[], int root, int heapSize){int leftchild = LEFT_CHILD(root);int rightchild = RIGHT_CHILD(root);int largestchild;if (leftchild < heapSize && data[leftchild] > data[root]){largestchild = leftchild;}else{largestchild = root;}if (rightchild < heapSize && data[largestchild] < data[rightchild]){largestchild = rightchild;}if (largestchild != root){swap(data[largestchild], data[root]);Max_Heapify(data, largestchild, heapSize);}}//构建最大二叉堆//从倒数第二层开始平衡最大堆即可void Build_Max_Heap(int data[], int heapSize){for (int i = (heapSize - 1) / 2; i >= 0; i--){Max_Heapify(data, i, heapSize);}}//堆排序//先构建最大二叉堆,然后再每次把根和最后一个节点替换,递归void HeapSort(int data[], int heapSize){Build_Max_Heap(data, heapSize);for (int i = heapSize - 1; i > 0; i--){swap(data[0], data[i]);Max_Heapify(data, 0, i);}}void Printn(int data[], int size){for (int i = 0; i < size; i++){cout << data[i] << " ";}cout << endl;}void Getn(int data[], int size){for (int i = 0; i < size; i++){cin >> data[i];}}int main(){int size;while (true){cout << "Please input the data szie(no more than " << ARRAY_SIZE << "):(0 to quit.)";cin >> size;if (0 == size){break;}Getn(data, size);HeapSort(data, size);Printn(data, size);}}


原创粉丝点击