堆排序

上一篇完成了堆的建立。在此基础上就可以实现堆排序。

思路:在此我们以实现从小到大排序为例。通过第一次的堆建立,生成的大顶堆的根结点值是最大值。然后我们将这个值放到某个地方可以不管它了(实际操作时都是放到整个堆的最后面,然后把最后面那个结点值放到最顶端),此时我们需要操作的元素个数已经少了一个,把剩下的元素再做一个建堆的操作(其实就是堆顶根结点元素的一个交换排列,其他节点的元素是之前已经排列好了的)。依次这个流程到最后。

注意:如果要做一个从大到小的排序,那么需要建立一个小顶堆,即最上面的元素是最小的。然后再堆排序的过程中会把最小的元素放到最后面,这样就实现了一个从大到小的排序。反之,从小到大排序,也是一样的道理。

下面看java实现的代码。综合了堆建立的代码。

package test;

/**
 *  @author hushunfeng
 *  
 *  大顶堆堆排序
 *  从小到大排序
 *  
 */
public class HeapSort {

    public static void main(String[] args) {
        int[] array = {10,20,30,40,50,60,70,80};
        //调整堆
        int temp = 0;
        for(int j=7;j>=0;j--) {
            createHeap(array,j+1);
            temp = array[0];
            array[0] = array[j];
            array[j] = temp;
        }
        for(int i=0;i<8;i++) {
            System.out.println(array[i]);
        }
    }
    
    /**
     * 将无序序列进行建堆
     * @param array 要建的堆
     */private static void createHeap(int[] array,int heapNum) {
        //heapNum = array.length;
        //根据建堆过程,先从第n/2个元素开始
        //它是最后一个非终端结点
        for(int i = heapNum/2-1; i >= 0; i--) {
            adjustNode(array,i,heapNum);
        }
    }
    
    /**
     * 
     * 将parentNode为根结点的子树调整成一个最大堆
     * @param array 要调整的无序序列,以数组的形式进行存储
     * @param parentNode  以此结点为根结点建立子树
     * 
     */
    //由于我们采用的是大顶堆,父结点要调整成大于子结点
    private static void adjustNode(int[] array,int parentNode,int nodeNum) {
        
        //数组长度
    //    int arrayNum = array.length;
        //跳动的父结点游标(在子树层数大于2层时适用)
        int parentIndex = parentNode;
        //左子结点
        int childLeftNode = 2*parentIndex+1;
        //右子结点
        int childRightNode = 2*parentIndex+2;
        //
        int maxNode;
        
        //将可能要跳动的父结点的数据缓存起来
        int temp = array[parentNode];
        
        //不断地跳动父结点的游标
        //防止右子结点不存在这种情况,不然会内存溢出
        while(childLeftNode<nodeNum) {
            //此种情况下已经不存在右结点了
            if(childLeftNode==nodeNum-1) {
                maxNode = childLeftNode;
            }
            else {
                //先比较左右子结点
                if(array[childLeftNode]>array[childRightNode])
                    maxNode = childLeftNode;
                else 
                    maxNode = childRightNode;
            }
            
            //比较,交换位置
            if(array[parentIndex]<array[maxNode]) {
                //将较大的子结点交换到父结点的位置
                array[parentIndex] = array[maxNode];
                array[maxNode] = temp;
                //往下移一层,再进行一次比较交换操作
                //原先的子结点现在变成了父结点
                parentIndex = maxNode;
                //下一层的子结点
                childLeftNode = 2*parentIndex+1;
                childRightNode = 2*parentIndex+2;
            }
            //如果判断出父结点为最大,直接跳出循环
            //无需再做交换动作,因为下面的元素都已经排成了大堆
            else break;
            
        }
        
    }
    
}

注:相比堆建立中,程序稍微进行了优化和修改。

修改后的第二版,考虑到从第二次建堆开始只需对根结点进行调整。

package test;

/**
 *  @author hushunfeng
 *  
 *  大顶堆堆排序
 *  从小到大排序
 *  
 */
public class HeapSort {

    public static void main(String[] args) {
        int[] array = {10,20,30,40,50,60,70,80};
        //先新建一个堆
        createNewHeap(array);
        //去除最后一个元素,将根结点进行重新调整
        int temp = 0;
        for(int j=7;j>=0;j--) {
            temp = array[0];
            array[0] = array[j];
            array[j] = temp;
            adjustNode(array,0,j);
        }
        for(int i=0;i<8;i++) {
            System.out.println(array[i]);
        }
    }
    
    /**
     * 将无序序列进行建堆
     * @param array 要建的堆
     */private static void createNewHeap(int[] array) {
        int heapNum = array.length;
        //根据建堆过程,先从第n/2个元素开始
        //它是最后一个非终端结点
        for(int i = heapNum/2-1; i >= 0; i--) {
            adjustNode(array,i,heapNum);
        }
    }
    
    /**
     * 
     * 将parentNode为根结点的子树调整成一个最大堆
     * @param array 要调整的无序序列,以数组的形式进行存储
     * @param parentNode  以此结点为根结点建立子树
     * 
     */
    //由于我们采用的是大顶堆,父结点要调整成大于子结点
    private static void adjustNode(int[] array,int parentNode,int nodeNum) {
        
        //数组长度
    //    int arrayNum = array.length;
        //跳动的父结点游标(在子树层数大于2层时适用)
        int parentIndex = parentNode;
        //左子结点
        int childLeftNode = 2*parentIndex+1;
        //右子结点
        int childRightNode = 2*parentIndex+2;
        //
        int maxNode;
        
        //将可能要跳动的父结点的数据缓存起来
        int temp = array[parentNode];
        
        //不断地跳动父结点的游标
        //防止右子结点不存在这种情况,不然会内存溢出
        while(childLeftNode<nodeNum) {
            //此种情况下已经不存在右结点了
            if(childLeftNode==nodeNum-1) {
                maxNode = childLeftNode;
            }
            else {
                //先比较左右子结点
                if(array[childLeftNode]>array[childRightNode])
                    maxNode = childLeftNode;
                else 
                    maxNode = childRightNode;
            }
            
            //比较,交换位置
            if(array[parentIndex]<array[maxNode]) {
                //将较大的子结点交换到父结点的位置
                array[parentIndex] = array[maxNode];
                array[maxNode] = temp;
                //往下移一层,再进行一次比较交换操作
                //原先的子结点现在变成了父结点
                parentIndex = maxNode;
                //下一层的子结点
                childLeftNode = 2*parentIndex+1;
                childRightNode = 2*parentIndex+2;
            }
            //如果判断出父结点为最大,直接跳出循环
            //无需再做交换动作,因为下面的元素都已经排成了大堆
            else break;
            
        }
        
    }
    
}
原文地址:https://www.cnblogs.com/hushunfeng/p/3912817.html