1. 程式人生 > >Java與算法之(8) - 堆排序

Java與算法之(8) - 堆排序

循環 public tar 最大 swap https rgs tool 技術分享

堆是一種特殊的完全二叉樹,其特點是所有父節點都比子節點要小,或者所有父節點都比字節點要大。前一種稱為最小堆,後一種稱為最大堆。

比如下面這兩個:

技術分享

那麽這個特性有什麽作用?既然題目是堆排序,那麽肯定能用來排序。想要用堆排序首先要創建一個堆,如果對4 3 6 2 7 1 5這七個數字做從小到大排序,需要用這七個數創建一個最大堆,來看代碼:

[java] view plain copy print?技術分享技術分享
  1. public class HeapSort {
  2. private int[] numbers;
  3. private int length;
  4. public HeapSort(int[] numbers) {
  5. this.numbers = numbers;
  6. this.length = numbers.length;
  7. }
  8. /**
  9. * 調整二叉樹
  10. * 如果父節點編號為x, 那麽左子節點的編號是2x, 右子節點的編號是2x+1
  11. * 節點編號從1開始, 對應數組中的索引是編號-1
  12. * @param nodeId 節點編號, 從1開始
  13. */
  14. public void adjust(int nodeId) {
  15. int swapId;
  16. int flag = 0; //是否需要繼續向下調整
  17. while(nodeId * 2 <= this.length && flag == 0) {
  18. //首先判斷它和左子節點的關系, 並用swapId記錄值較小的節點編號(最大堆是記錄較大的)
  19. int index = nodeId - 1; //節點對應數組中數字的索引
  20. int leftChild = nodeId * 2 - 1; //左子節點對應數組中數字的索引
  21. int rightChild = nodeId * 2; //右子節點對應數組中數字的索引
  22. if(numbers[index] < numbers[leftChild]) {
  23. swapId = nodeId * 2;
  24. } else {
  25. swapId = nodeId;
  26. }
  27. //如果有右子節點, 再與右子節點比較
  28. if(nodeId * 2 + 1 <= this.length) {
  29. if(numbers[swapId - 1] < numbers[rightChild])
  30. swapId = nodeId * 2 + 1;
  31. }
  32. //如果最小的節點編號不是自己, 說明子節點中有比父節點更小的
  33. if(swapId != nodeId) {
  34. swap(swapId, nodeId);
  35. nodeId = swapId;
  36. } else {
  37. flag = 1;
  38. }
  39. }
  40. }
  41. /**
  42. * 交換兩個節點的值
  43. * @param nodeId1
  44. * @param nodeId2
  45. */
  46. public void swap(int nodeId1, int nodeId2) {
  47. int t = numbers[nodeId1 - 1];
  48. numbers[nodeId1 - 1] = numbers[nodeId2 - 1];
  49. numbers[nodeId2 - 1] = t;
  50. }
  51. /**
  52. * 創建最大堆
  53. */
  54. public void createMaxHeap() {
  55. //從最後一個非葉節點到第一個節點依次向上調整
  56. for(int i = this.length / 2; i >= 1; i--) {
  57. adjust(i);
  58. }
  59. }
  60. public static void main(String[] args) {
  61. int[] numbers = new int[] { 4, 3, 6, 2, 7, 1, 5 };
  62. for(int x = 0; x < numbers.length; x++) {
  63. System.out.print(numbers[x] + " ");
  64. }
  65. System.out.println();
  66. HeapSort heap = new HeapSort(numbers);
  67. heap.createMaxHeap();
  68. }
  69. }

對本例中的數列,從this.length / 2到1,共執行了三輪循環。

第一輪:

技術分享

第二輪:

技術分享
第三輪:

技術分享

調整完成後,當前的二叉樹已經符合最大堆的特性,可以用來從小到大排序。堆排序的原理是,交換堆頂和最後一個節點的數字,即把最大的數字放到數組最後,然後對除了最大數的前n-1個數從新執行調整過程,使其符合最大堆特性。重復以上過程直到堆中只剩下一個數字。

[java] view plain copy print?技術分享技術分享
  1. public void sort() {
  2. while(this.length > 1) {
  3. swap(1, this.length);
  4. this.length--;
  5. adjust(1);
  6. }
  7. for(int x = 0; x < numbers.length; x++) {
  8. System.out.print(numbers[x] + " ");
  9. }
  10. }

堆排序的時間復雜度和快速排序的平均時間復雜度一樣,是O(nlogn)。

Java與算法之(8) - 堆排序