堆排序：对数组片段排序

堆排序：对数组片段排序

部分代码参考《算法》第四版

package algorithms.sort;
 
public class HeapSort {
    private static boolean less(int[] a, int i, int j) {
    	if(a[i] < a[j])
    		return true;
    	return false;
    }
    private static void exch(int[] a, int i, int j) {
    	int temp = a[i];
    	a[i] = a[j];
    	a[j] = temp;
    }
    //一个问题有多种表示方式，都能反映问题的本质
    //关键在于理解堆结构的索引情况，以及数组的元素位置。
    //使用数组a的第一个元素，原地操作，不借助辅助数组
    //堆结构中的元素从1开始索引，M是堆结构最后一个元素的索引，也就是堆结构长度，也是数组长度。k是堆中第k个元素的索引
    //数组a是从0开始索引的，如果将数组从左到右顺序放入堆中，那么堆k处放置数组元素a[k-1]，a[M-1]放置数组最后一个元素，要求M传入a.length
    private static void sink(int[] a, int k, int M) {//把数组a当做是无序堆
    	while(2 * k <= M) { //k,j都是堆的位置索引，   2 * k > M 时，k处元素没有子节点，找到一个最大的k，使得2 * k <= M
    		int j = 2 * k;  //当前j是左节点，但是a[j]却是索引到右节点的元素，左节点索引j实际存放a[j-1]这个元素
    		if(j < M && less(a, j - 1, j)) j++; //j < M即有右节点的情况。j处取左右子节点的较大值的索引
    		if(less(a, j - 1, k - 1))  break; //根节点索引k实际存放a[k-1]这个元素
    		exch(a, k - 1, j - 1);
    		k = j;  
    	}
    }
    
    //对数组片段进行排序
    //将数组索引lo ~ hi区间转到堆的1 ~ M之间
    private static void sink2(int[] a, int k, int M, int lo) {//把数组a当做是无序堆
    	while(2 * k <= M) {  
    		int j = 2 * k;   
    		if(j < M && less(a, j - 1 + lo, j + lo)) j++;  
    		if(less(a, j - 1 + lo, k - 1 + lo))  break;  
    		exch(a, k - 1 + lo, j - 1 + lo);
    		k = j;   
    	}
    } 
  //错误案例 
    private static void sink3(int[] a, int k, int M) {//把数组a当做是无序堆
    	while(2 * k <= M) { //k,j都是堆的位置索引，   2 * k > M 时，k处元素没有子节点，找到一个最大的k，使得2 * k <= M
    		int j = 2 * k;  //当前j是左节点，但是a[j]却是索引到右节点的元素，左节点索引j实际存放a[j-1]这个元素
    		//j要么移动，要么不移动
    		if(j < M && less(a, j, j - 1)) j--; //其实j并没有移动，未考虑堆上j处小于j+1处元素情况
    		if(less(a, j, k - 1))  break; //根节点索引k实际存放a[k-1]这个元素
    		exch(a, k - 1, j);
    		k = j + 1; 
    	}
    } 
    public static void sort(int[] a)  {
    	int M = a.length;
    	for(int k = M/2; k >= 1; k--)
    		sink(a, k, M);
    	while(M > 1) {
    		exch(a, 0, --M); //a.length-1是最后一个元素
    		sink(a, 1, M);
    	}
    } 
    
    public static void sort2(int[] a, int lo, int hi)  {
    	if(lo < 0 || hi < 0 || lo >= hi) return;
    	int M = hi - lo + 1;
    	for(int k = M/2; k >= 1; k--)
    		sink2(a, k, M, lo); 
    	while(M > 1) { //或者使用hi > lo判断
    		exch(a, lo, hi--);  
    		sink2(a, 1, --M, lo);
    		 
    	}
    }
	public static void main(String[] args) {
		int[] a = {12,9,28,13,4,5,11,1,71}; 
		for(int v : a)
        	System.out.print(v + " ");
		System.out.println();
        sort(a);
        for(int v : a)
        	System.out.print(v + " ");
        
        System.out.println();
        int[] b = {12,2,9,28,13,4,5,90,11,1,71};   
        for(int v : b)
        	System.out.print(v + " ");
		System.out.println();
        sort(b);
        for(int v : b)
        	System.out.print(v + " ");  
       
        System.out.println();
        int[] c = {12,2,9,28,13,4,5,90,11,1,71};  
        for(int v : c)
        	System.out.print(v + " ");
		System.out.println();
        sort2(c, 3, 8);
        for(int v : c)
        	System.out.print(v + " ");  
         
        System.out.println();
        int[] e = {-1,-9,0,1,45,-1,-9,12,2,9,28,13,4,5,90,11,1,71};  
        for(int v : e)
        	System.out.print(v + " ");
		System.out.println();
		//使e[3]~e[8]有序
		sort2(e, 3, 8);
        for(int v : e)
        	System.out.print(v + " ");  
        
		/*
		//测试sink() 和sink2()
        System.out.println();
        int[] d = {-1,-9,0,1,45,-1,-9,12,2,9,28,13,4,5,90,11,1,71}; 
        for(int v : d)
        	System.out.print(v + " ");
		System.out.println();
		for(int i = 1; i <= d.length; i++)
		  sink(d, i, d.length); 
		
		
		System.out.println();
        int[] e = {-1,-9,0,1,45,-1,-9,12,2,9,28,13,4,5,90,11,1,71};  
        for(int v : e)
        	System.out.print(v + " ");
		System.out.println();
		for(int i = 1; i <= e.length; i++)
		   sink2(e, i, e.length, 0); */
	}

}

输出：

12 9 28 13 4 5 11 1 71 
1 4 5 9 11 12 13 28 71 
12 2 9 28 13 4 5 90 11 1 71 
1 2 4 5 9 11 12 13 28 71 90 
12 2 9 28 13 4 5 90 11 1 71 
12 2 9 4 5 11 13 28 90 1 71 
-1 -9 0 1 45 -1 -9 12 2 9 28 13 4 5 90 11 1 71 
-1 -9 0 -9 -1 1 2 12 45 9 28 13 4 5 90 11 1 71