Java实现堆的封装,进行插入,调整,删除堆顶以完成堆排序实例

简介


堆对于排序算法是一个比较常用的数据结构,下面我就使用Java语言来实现这一算法


首先,我们需要知道堆的数据结构的形式,其实就是一个特殊的二叉树。但是这个二叉树有一定的特点,除了是完全二叉树以外,对于最大堆而言,堆顶元素的值是最大的,而且对于堆的每一个子树也是一个小一号的最大堆;同样对于最小堆,性质相反就可以了。


我以最大堆为例:
要实现堆的初始化操作,就是先按照给定的元素创建一棵完全二叉树,然后从末尾节点进行不断地调整的过程。调整的原则是:比较要进行放置的当前节点与其父节点的数值的大小,若要进行放置的当前节点的值小于其父节点,那么当前节点所在位置符合最大堆的定义,要进行放置的当前节点放在此处是比较合适的;如果要进行放置的当前节点的值大于其父节点的值,那说明放在当前节点是不合适的,那么就需要将当前节点的值与其父节点的值进行交换,然后原父节点变为新的要进行放置的当前节点。循环比较;终止条件就是当前节点没有父节点,但此时的调整也许并没有结束,我们只需要让堆顶元素为要插入的值即可。至此,最大堆的插入和调整过程结束。
代码如下:

public boolean insert(int x){
        if(currentSize==MAXSIZE){
            System.out.println("Sorry,this heap is full!");
            return false;
        }
        //如果堆不满的话
        currentSize++;
        int flag=currentSize-1;
        while(flag>0){
            int parent=(flag-1)/2;
            if(heap[parent]>x){
                heap[flag]=x;
                return true;
            }else{
                heap[flag]=heap[parent];
                flag=parent;
            }
        }
        heap[0]=x;
        return true;
    }

siftDown过程:给定一个节点的位置,对其进行调整,使之符合最大堆的定义,这个过程就是我们要实现的过程。调整原则如下:
对于当前节点i而言,其孩子节点的下标满足左节点为2*i+1,右节点为2*i+2;在进行调整的过程中,只需要比较当前节点与其子节点中最大的节点进行调整即可。具体的代码逻辑可在代码中看到:

public void siftDown(int flag){
        int want=flag;
        int x=heap[flag];

        while(want<currentSize){
            int lChild=2*want+1;
            int rChild=2*want+2;
            int MAXChildNumber;
            if(lChild>currentSize){  //没有孩子节点
                heap[want]=x;
            }else{                   //有两个孩子节点
                if(lChild<currentSize){
                    MAXChildNumber=heap[lChild]>heap[rChild]?lChild:rChild;
                }else{
                    MAXChildNumber=lChild;
                }
                if(heap[MAXChildNumber]<x){
                    heap[want]=x;return;
                }else{
                    heap[want]=heap[MAXChildNumber];
                    want=MAXChildNumber;
                }
            }
        }

    }

堆顶元素的删除,我们对堆的操作基本桑就是为了获得这个堆的最值,那么毫无疑问,堆顶元素就是我们要研究的对象。下面是代码逻辑:

public int deleteTop(){
        if(currentSize<0){
            System.out.println("Sorry, this heap is empty!");
            return -1;
        }
        int target=heap[0];
        int substitute=heap[currentSize-1];
        this.currentSize--;
        heap[0]=substitute;
        siftDown(0);
        return target;
    }

下面是详细的代码

package test.maxHeap;

public class MaxHeap {

    private int []heap ;
    private int currentSize;
    private static int MAXSIZE ;

    public MaxHeap(int n){
        heap=new int[n];
        currentSize=0;
        MAXSIZE=n;
    }

    public boolean insert(int x){
        if(currentSize==MAXSIZE){
            System.out.println("Sorry,this heap is full!");
            return false;
        }
        //如果堆不满的话
        currentSize++;
        int flag=currentSize-1;
        while(flag>0){
            int parent=(flag-1)/2;
            if(heap[parent]>x){
                heap[flag]=x;
                return true;
            }else{
                heap[flag]=heap[parent];
                flag=parent;
            }
        }
        heap[0]=x;
        return true;
    }

    public void siftDown(int flag){
        int want=flag;
        int x=heap[flag];

        while(want<currentSize){
            int lChild=2*want+1;
            int rChild=2*want+2;
            int MAXChildNumber;
            if(lChild>currentSize){  //没有孩子节点
                heap[want]=x;
            }else{                   //有两个孩子节点
                if(lChild<currentSize){
                    MAXChildNumber=heap[lChild]>heap[rChild]?lChild:rChild;
                }else{
                    MAXChildNumber=lChild;
                }
                if(heap[MAXChildNumber]<x){
                    heap[want]=x;return;
                }else{
                    heap[want]=heap[MAXChildNumber];
                    want=MAXChildNumber;
                }
            }
        }

    }

    public int deleteTop(){
        if(currentSize<0){
            System.out.println("Sorry, this heap is empty!");
            return -1;
        }
        int target=heap[0];
        int substitute=heap[currentSize-1];
        this.currentSize--;
        heap[0]=substitute;
        siftDown(0);
        return target;
    }

}

好了,编码已经完成。下面我们就要检验一下是否正确吧。

public class MaxHeapTest {

    public static void main(String []args){
        MaxHeap maxHeap=new MaxHeap(7);
        for(int i=1;i<=7;i++){
            maxHeap.insert(i);
        }
        for(int i=0;i<7;i++){
            System.out.print(maxHeap.deleteTop()+"   ");
        }
        System.out.println("\n");
    }
}

接下来是程序的运行结果:

7   6   5   4   3   2   1   
//可见,对于最大堆,删除堆顶的操作实际上就是完成了对堆的排序任务,也证明了我们的代码是正确的

总结:
堆的操作很重要,我们更要学会对于堆的应用,这样的数据结构才能使得程序的运行更加的高效和流畅。对于最小堆,我们只需要在插入方法,sift方法内稍加修改即可(也就是将值的代销变换关系进行调整)。这样就同样能实现最小堆的创建和相关的操作了。
代码中可能存在不太恰当地地方,希望大家予以批评指正,期待与你们共同进步!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值