对于一个int数组,请编写一个堆排序算法,对数组元素排序。
给定一个int数组A及数组的大小n,请返回排序后的数组。
测试样例:
[1,2,3,5,2,3],6
[1,2,2,3,3,5]
AC代码:
import java.util.*;
public class HeapSort {
public int[] heapSort(int[] A, int n) {
// write code here
// 创建最大堆
for (int i=(n/2-1); i>=0; i--) {
//忽略叶子节点,从拥有叶子节点的下编号最大的节点开始,一直调整到根结点
MaxHeap(A, i, n);
}
for (int i=0; i<n-1; i++) {
// 把储存最大值的根结点调到数组尾部
swap(A, 0, n-1-i);
// 去掉尾节点,堆长度减1,调整最大堆,只需要重新调整根结点
MaxHeap(A, 0, n-1-i);
}
return A;
}
public void MaxHeap(int A[], int index, int len) {
// index为待调整节点,maxIndex为index的子节点; 若交换后子节点仍有其子节点,需要继续往下调整
for (int maxIndex=2*index+1; maxIndex<len; maxIndex=2*maxIndex+1) {
// 判断右孩子是否存在,且是否比左孩子大,选出最大孩子节点
if ((maxIndex+1)<len && A[maxIndex+1]>A[maxIndex]) {
maxIndex ++;
}
// 父与子节点数值判断
if (A[index] > A[maxIndex]) {
break;
} else {
swap(A, index, maxIndex);
index = maxIndex;
}
}
}
public void swap(int[] A, int i, int j) {
//不用开辟新内存
A[i] = A[i] + A[j];
A[j] = A[i] - A[j];
A[i] = A[i] - A[j];
}
}