构建堆的时间复杂度为O(n),而第I次调整堆的时间复杂度为O(logi),因此,无论什么情况下时间复杂度都为O(nlogn)。
首先,对数组从n/2处开始进行创建堆。大顶堆就是顶点总是大于它的子节点,而小顶堆就是定点总是小于它的子节点。
因此,构建时,对节点与他的孩子进行比较,如果创建的是大顶堆,那么就把最大的孩子跟自己进行比较,比自己大,就进行替换。依次类推。
最后把最顶点的与最后一个元素进行替换,在进行堆的调整,此时的调整,相当于调整最顶点的元素。
void heapSort(int *arr,int length){ int i; for(i=length/2; i>0; i--){ sort(arr,i,length); } for(i=length; i>0; i--){ int k = arr[0]; arr[0] = arr[i-1]; arr[i-1] = k; sort(arr,1,i-1); } } void sort(int *arr,int s,int length){ int temp,i; temp = arr[s-1]; for(i=2*s; i<=length; i*=2 ){ if(i<length && arr[i-1] < arr[i]) i++; if(temp >= arr[i-1]) break; arr[s-1] = arr[i-1]; s = i; } arr[s-1] = temp; }
#include <stdio.h> #include <stdlib.h> #include <time.h> int arrtest1[10] = {3,4,7,8,0,9,1,2,6,5}; int arrtest2[10] = {0,1,2,3,4,5,6,7,8,9}; int arrtest3[10] = {9,8,7,6,5,4,3,2,1,0}; void copy(int *from,int *arr,int length); void print(int *arr,int length); void heapSort(int *arr,int length); void sort(int *arr,int s,int length); int main(){ clock_t start,end; int Arr[10]; int i; copy(arrtest1,Arr,10); print(Arr,10); heapSort(Arr,10); print(Arr,10); start = clock(); for(i=0;i<100000;i++){ copy(arrtest1,Arr,10); //print(Arr,10); heapSort(Arr,10); //print(Arr,10); } end = clock(); printf("first test:%d\n",end-start); start = clock(); for(i=0;i<100000;i++){ copy(arrtest2,Arr,10); //print(Arr,10); heapSort(Arr,10); //print(Arr,10); } end = clock(); printf("first test:%d\n",end-start); start = clock(); for(i=0;i<100000;i++){ copy(arrtest3,Arr,10); //print(Arr,10); heapSort(Arr,10); //print(Arr,10); } end = clock(); printf("first test:%d\n",end-start); getchar(); return 0; } void heapSort(int *arr,int length){ int i; for(i=length/2; i>0; i--){ sort(arr,i,length); } for(i=length; i>0; i--){ int k = arr[0]; arr[0] = arr[i-1]; arr[i-1] = k; sort(arr,1,i-1); } } void sort(int *arr,int s,int length){ int temp,i; temp = arr[s-1]; for(i=2*s; i<=length; i*=2 ){ if(i<length && arr[i-1] < arr[i]) i++; if(temp >= arr[i-1]) break; arr[s-1] = arr[i-1]; s = i; } arr[s-1] = temp; } void copy(int *from,int *arr,int length){ int i; for(i=0;i<length;i++){ arr[i] = from[i]; } } void print(int *arr,int length){ int i; for(i=0;i<length;i++){ printf("%d ",arr[i]); } printf("\n"); }