在介绍二叉堆的时候,对于deleteMin()操作(删除最下元素)的时候,我们知道时间复杂度为O(logN)。而创建二叉堆的时候的时间复杂度为O(N)。所以用二叉堆实现的的堆排序的时间复杂度为O(NlogN)。
代码:
public class heapSort {
public void sort(int[] nums) {
buildHeap(nums);
}
//此时的开始位置是从0的位置开始的
private void buildHeap(int[] nums) {
for(int i = nums.length/2-1;i>=0;i--) {
up(nums,i,nums.length);
}
//进行删除操作
for(int i = nums.length-1;i>=0;i--){
deleteMax(nums,i);
}
}
private void up(int[] nums,int index,int length) {
//需要一个最小位置
int child=0;
//需要一个临时变量
int temp;
//找到结束的条件
for(temp = nums[index];getChild(index)<length;index = child) {
child = getChild(index);
//要建立大根堆,所以找左右孩子的最大值
if(child+1<length && nums[child+1]>nums[child]) {
child +=1;
}
//index的位置要是最大值
if(temp < nums[child]) {
nums[index] = nums[child];
}
else {
break;
}
}
//结束循环后一定要为index的位置进行赋值。
nums[index] = temp;
}
//获取左孩子位置
private int getChild(int root) {
return 2*root+1;
}
private void deleteMax(int[] nums,int length) {
//先把最后的这个元素和第一个元素进行交换
int temp = nums[0];
nums[0] = nums[length];
nums[length] = temp;
//然后对大根堆进行调整
up(nums,0,length);
}
public static void main(String[] args) {
int[] nums = {5,9,7,6,88,45,62,1,7};
heapSort sort = new heapSort();
sort.sort(nums);
for(int i = 0;i<nums.length;i++) {
System.out.print(nums[i]+",");
}
}
}