算法-java代码实现堆排序

---

堆排序

第7节 堆排序练习题

对于一个int数组，请编写一个堆排序算法，对数组元素排序。

给定一个int数组A及数组的大小n，请返回排序后的数组。

测试样例：

[1,2,3,5,2,3],6

[1,2,2,3,3,5]

 

Java (javac 1.7)

代码自动补全

 

 

 

 

 

1

import java.util.*;

2

3

public class HeapSort {

4

    public int[] heapSort(int[] A, int n) {

5

        int lastIndex = n - 1;

6

        buildMaxHeap(A, lastIndex);//建立最大堆

7

        while(lastIndex > 0){

8

            swap(A, 0, lastIndex);

9

            if(--lastIndex == 0)//只剩一个元素，就不用调整堆了，排序结束

10

                break;

11

            adjustHeap(A,0,lastIndex);    

12

        }

13

        return A;

14

    }

15

16

    public void buildMaxHeap(int[] arr, int lastIndex) {

17

        // 从最后一个元素的父节点开始进行调整，一直调整到根节点结束

18

        int j = (lastIndex - 1) / 2;

19

        while (j >= 0) {

20

            int rootIndex = j;

21

            adjustHeap(arr, rootIndex, lastIndex);

22

            j--;

23

        }

24

    }

25

26

    public void adjustHeap(int[] arr, int rootIndex, int lastIndex) {//从根节点开始往下调整

27

        int biggerIndex = rootIndex;

28

        int leftChildIndex = rootIndex * 2 + 1; 

29

        int rightChildIndex = rootIndex * 2 + 2;

30

31

        if(rightChildIndex <= lastIndex){//如果右孩子存在，则左孩子一定存在

32

            if(arr[rightChildIndex] > arr[rootIndex] || arr[leftChildIndex] > arr[rootIndex]){

33

                //将子节点更大的元素下标赋值给biggerIndex

34

                biggerIndex = arr[rightChildIndex] > arr[leftChildIndex]?rightChildIndex:leftChildIndex;

35

            }

36

        }

37

        else if(leftChildIndex <= lastIndex){//保证左孩子存在，且不能越界

38

            if(arr[leftChildIndex] > arr[rootIndex]){

39

                biggerIndex = leftChildIndex;

40

            }

41

        }

42

        if(biggerIndex != rootIndex){

43

            swap(arr, biggerIndex, rootIndex);

44

            adjustHeap(arr, biggerIndex, lastIndex);

45

        }

46

47

    }

48

49

    public void swap(int[] arr, int biggerIndex, int rootIndex) {

50

        int temp = arr[rootIndex];

51

        arr[rootIndex] = arr[biggerIndex];

52

        arr[biggerIndex] = temp;

53

    }

54

}

 

 

您的代码已保存
答案正确:恭喜！您提交的程序通过了所有的测试用例