堆排序算法主要分为初始化堆和交换最大值重建堆的过程,下面记录下它的代码实现:
public class Heap {
public Heap() {
}
public static void sort(Double[] pq) {
//System.out.println(pq.length);
int N=pq.length;
for(int i=N/2;i>=1;i--) sink(pq,i,N);// 6 12
while(N>1) {
exch(pq,1,N--);
sink(pq,1,N);
}
}
private static void sink(Double[] pq,int k,int n) {
while(2*k<=n) {
int j=2*k;//12
if(j<n&&pq[j-1]<pq[j+1-1]) j++;
if(pq[k-1]>=pq[j-1]) break;
exch(pq,k,j);
k=j;
}
}
private static void exch(Double[] a,int b,int c) {
double temp=a[b-1];
a[b-1]=a[c-1];
a[c-1]=temp;
}
}
时间复杂度
堆排序算法包括构建堆的过程和交换最大值重建堆的过程。
构建堆的过程:令k为堆的高度,i表示第几层,构建堆进行的是下沉操作,比较和交换是从倒数第二层开始的,一直往上到第一层,每次交换会比较第n、2n、2n+1位置的元素。因此可得每层所需时间为:
S=[2^(i-1)]*(k-i)。
其中2^(i-1)表示该层有多少个元素,k-i表示该层上所需的比较次数。
总的时间即S=[2(k-2)]*1+[2(k-3)]*2+[2(k-4)]*3+…[2(0)]*k-1。
由2S-S=2k-1+2k-2+2k-3…+2-k+1,等比求和即S=2k-k+1,因为k为堆高度等于logn,所以S=n-logn+1<n,即构建堆的时间复杂度是O(n)。
交换最大值重建堆的过程:堆高为logn,每次交换最大值n自减,即S=logn+log(n-1)+log(n-1)+…log1<nlogn,即交换最大值重建堆的时间复杂度是O(nlgn)。
总时间复杂度T(n)=O(n)+O(nlgn)=O(nlgn)。
空间复杂度
因为堆排序是就地排序,空间复杂度为常数:O(1)。