数据结构-堆的实现

标签：

3.堆是一棵完全二叉树。

堆有两种：最大堆和最小堆。

最小堆中每个节点的优先级小于或者等于它的子节点；最大堆则相反，每个节点的优先级都大于或者等于它的子节点。

如下图：

本文只着重讲最大堆吧，最小堆是一样的。

堆的大小是提前知道的，在java集合中堆是通过ArrayList数组实现的：

1.根节点位置：根节点的数据总是在数组的位置[0]
2.节点的父节点位置：假设一个非根节点的数据在数组中的位置[i]，那么它的父节点总是在位置[(i-1)/2]
3.节点的孩子节点位置：假设一个节点的数据在数组中的位置为[i]，那么它的孩子（如果有）总是在下面的这两个位置：左孩子在[2*i+1]，右孩子在[2*i+2]

在堆中主要是插入和删除节点的操作，这两种操作无论是哪一种，完成之后都还是一个堆，操作时要进行堆的调整，使其是个新堆。

一，添加一个新节点：（不断地和父节点比较）

插入的思路是这样的：当插入一个元素时，先将这个元素插入到队列尾，然后将这个新插入的元素和它的父节点进行优先权的比较，如果比父节点的优先权要大，则和父节点呼唤位置，然后再和新的父节比较，直到比新的父节点优先权小为止

二，删除一个节点（不断比较左右孩子节点大小，大的上移）

从堆中删除优先权最大的元素的思路是将队列尾的元素值赋给根节点，队列为赋 值为null。然后检查新的根节点的元素优先权是否比左右子节点的元素的优先权大，如果 比左右子节点的元素的优先权小，就交换位置，重复这个过程，直到秩序正常。

heap.java实现

//最大堆
import java.util.ArrayList;
public class Heap<E extends Comparable>{
	private ArrayList<E> list=new ArrayList<E>();//用数组实现堆
	
    public Heap(){}
    public Heap(E[] objects){
    	for(int i=0;i<objects.length;i++){
    		add(objects[i]);
    	}
    }
    
    public void add(E newObject){//添加一个元素
    	list.add(newObject);
    	int currentIndex=list.size()-1;
    	
    	while(currentIndex>0){
    		int parentIndex=(currentIndex-1)/2;//找到该结点的父结点
    		if(list.get(currentIndex).compareTo(list.get(parentIndex))>0){//与父节点比较
    			//如果当前结点的值大于父结点就交换位置
    			E temp=list.get(currentIndex);
    			list.set(currentIndex, list.get(parentIndex));
    			list.set(parentIndex, temp);   			
    		}
    		else
    			break;
    		currentIndex=parentIndex;
    	}    	
    }
    
    public E remove(){//删除并返回根结点,堆的特点是移除了根结点后还是堆
    	if(list.size()==0) return null;
    	
    	E removeObject=list.get(0);
    	list.set(0, list.get(list.size()-1));//把最后一个结点放在根结点的位置
    	list.remove(list.size()-1);
    	
    	int currentIndex=0;
    	while(currentIndex<list.size()){
    		int leftChildIndex=2*currentIndex+1;
    		int rightChildIndex=2*currentIndex+2;//左右孩子结点的坐标
    		
    		if(leftChildIndex>=list.size())break;
    		//比较左右孩子的值，使maxIndex指向值大的结点
    		 int maxIndex=leftChildIndex;
    		 if(rightChildIndex<list.size()){
    			 if(list.get(maxIndex).compareTo(list.get(rightChildIndex))<0){
    				 maxIndex=rightChildIndex;
    			 }
    		 }
    		 //如果当前结点的值小于其左右孩子中的大的值，就交换两个结点
    		 if(list.get(currentIndex).compareTo(list.get(maxIndex))<0){
    	          E temp=list.get(maxIndex);
    	          list.set(maxIndex, list.get(currentIndex));
    	          list.set(currentIndex, temp);
    	          currentIndex=maxIndex;
    	    	}
    		 else
    			 break;
    	}
    	
    	return removeObject;   	
    	
    }
    
    public int getSize(){
    	return list.size();
    }
    
}

数据结构-堆的实现

标签：

原文地址：http://blog.csdn.net/tuke_tuke/article/details/50357939