如何在 Java 中对数组执行堆排序?


以下是堆排序(最大堆)的算法。

步骤 1 − 在堆的尾部创建一个新节点。

步骤 2 − 给新节点分配新值。

步骤 3 − 将此子节点的值与其父节点进行比较。

步骤 4 − 如果父节点的值小于子节点,则交换它们的位置。

步骤 5 − 重复步骤 3 和 4,直至堆属性成立。

示例

import java.util.Arrays;
import java.util.Scanner;

public class Heapsort {
   public static void heapSort(int[] myArray, int length) {
      int temp;
      int size = length-1;
      for (int i = (length / 2); i >= 0; i--) {
         heapify(myArray, i, size);
      };
      for(int i= size; i>=0; i--) {
         temp = myArray[0];
         myArray[0] = myArray[size];
         myArray[size] = temp;
         size--;
         heapify(myArray, 0, size);
      }
      System.out.println(Arrays.toString(myArray));
   }
   public static void heapify (int [] myArray, int i, int heapSize) {
      int a = 2*i;
      int b = 2*i+1;
      int largestElement;
      if (a<= heapSize && myArray[a] > myArray[i]) {
         largestElement = a;
      } else {
         largestElement = i;
      }
      if (b <= heapSize && myArray[b] > myArray[largestElement]) {
         largestElement = b;
      }
      if (largestElement != i) {
         int temp = myArray[i];
         myArray[i] = myArray[largestElement];
         myArray[largestElement] = temp;
         heapify(myArray, largestElement, heapSize);
     }
   }
   public static void main(String args[]) {
      Scanner scanner = new Scanner(System.in);
      System.out.println("Enter the size of the array :: ");
      int size = scanner.nextInt();
      System.out.println("Enter the elements of the array :: ");
      int[] myArray = new int[size];
      for(int i=0; i<size; i++) {
         myArray[i] = scanner.nextInt();
      }
      heapSort(myArray, size);
   }
}

输出

Enter the size of the array ::
5
Enter the elements of the array ::
45
125
44
78
1
[1, 44, 45, 78, 125]

更新于:2020 年 2 月 19 日

590 次浏览

开启你的职业

完成课程后获得认证

开始学习
广告