算法一直是自己很弱的一环,最近争取有时间就投入上来。
直接贴上代码便于以后复习。
public class HeapSort {
public void heapAdjust(int[] heap, int len, int p) {
int lchild = 2 * p;
int rchild = 2 * p + 1;
int largest = p;
if(lchild <= len && heap[largest] < heap[lchild]) {
largest = lchild;
}
if(rchild <= len && heap[largest] < heap[rchild]) {
largest = rchild;
}
if(largest != p) {
swap(heap, largest, p);
heapAdjust(heap, len, largest);
}
}
public void swap(int[] a, int i, int j) {
int tmp = a[i];
a[i] = a[j];
a[j] = tmp;
}
public void buildHeap(int[] heap, int len) {
int p = len / 2;
for(int i = p; i >= 1; i--) {
heapAdjust(heap, len, i);
}
}
public void heapSort(int[] heap) {
int len = heap.length - 1;
buildHeap(heap, len);
for(int i = len; i >= 2; --i) {
swap(heap, i, 1);
heapAdjust(heap, i - 1, 1);
}
}
public static void main(String[] args) {
int[] a = {0, 18, 8, 5, 10, 6, 13, 14, 7, 5, 17};
HeapSort hs = new HeapSort();
hs.heapSort(a);
hs.pOut(a, a.length - 1);
}
public void pOut(int[] a, int k) {
for(int i = 1; i <= k; i++) {
System.out.print(a[i] + " ");
}
System.out.println();
}
}