十大經典排序算法——堆排序(Heap Sort)

文章目錄

算法

堆是一種優先隊列,兩種實現,最大堆和最小堆,由於這裏是升序,以最大堆來介紹。

代碼

package sort;
import java.util.Arrays;
/**
 * @author Heroin X
 * @date 2020/3/30 15:46
 */
public class HeapSort {
    public static void main(String[] args) {
        int[] arr = new int[]{3,5,1,4,7,2,6};
        new HeapSort().heapSort(arr);
        System.out.println(Arrays.toString(arr));
    }

    /**
     * 堆排序
     * @param arr
     */
    public void heapSort(int[] arr){
        int len = arr.length;
        for (int i = (len - 2)/2; i >=0; i--){
            downAdjust(arr, i, len);
        }
        for (int i = len - 1; i > 0; i--){
            int tmp =arr[i];
            arr[i] = arr[0];
            arr[0] = tmp;
            downAdjust(arr,0, i);
        }
    }

    /**
     * 下沉 -最大堆
     * @param arr
     * @param parIndex
     * @param len
     */
    private void downAdjust(int[] arr, int parIndex, int len) {
        int tmp = arr[parIndex];
        int chl = parIndex * 2 + 1;
        while (chl < len){
            if(chl + 1 < len && arr[chl+1] > arr[chl]){
                chl++;
            }

            if (tmp >= arr[chl]){
                break;
            }
            // 不需要真正的交換,單向賦值即可
            arr[parIndex] = arr[chl];
            // 新的parentIndex
            parIndex  = chl;
            // 新的childrenIndex
            chl = 2 * parIndex + 1;
//            chl = 2 * chl + 1;
        }
        arr[parIndex] = tmp;
    }
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章