import java.util.*;/*** 实现的最大堆的插入和删除操作
*
*@authorYou*/
public classHeap {/*** 删除堆中位置是index处的值 操作原理是:当删除节点的数值时,原来的位置就会出现一个孔
* 填充这个孔的方法就是,把最后的叶子的值赋给该孔,最后把该叶子删除
*
*@paramheap
* 大顶堆*/
public static void delete(List heap, intindex) {//把最后的一个叶子的数值赋值给index位置
heap.set(index, heap.get(heap.size() - 1));//下沉操作//heapDown2(heap, index);
heapDown(heap, index); //节点下沉//把最后一个位置的数字删除
heap.remove(heap.size() - 1);
}/*** 节点下沉递归实现 删除一个堆中一个数据的时候,根据堆的性质,应该把相应的位置下移,才能保持住堆性质不变
*
*@paramheap
* 保持最大堆元素的数组
*@paramindex
* 被删除的那个节点的位置*/
public static void heapDown(List heap, intindex) {//因为第一个位置存储的是空值,不在考虑之内
int n = heap.size() - 2;//记录最大的那个儿子节点的位置
int child = -1;//2*index>n说明该节点没有左右儿子节点了,那么就返回
if (2 * index >n) {return;
}//如果左右儿子都存在
else if (2 * index
child = 2 *index;//如果左儿子小于右儿子的数值,取右儿子的下标
if ((Integer) heap.get(child) < (Integer) heap.get(child + 1)) {
child++;
}
}//如果只有一个儿子(左儿子节点)
else if (2 * index ==n) {
child= 2 *index;
}if ((Integer) heap.get(child) >(Integer) heap.get(index)) {//交换堆中的child,和index位置的值
swap(heap, child, index);//完成交换后递归调用,继续下降
heapDown(heap, child);
}
}//非递归实现
public static void heapDown2(List heap, intindex) {int child = 0;//存储左儿子的位置
int temp =(Integer) heap.get(index);int n = heap.size() - 2;//如果有儿子的话
for (; 2 * index <= n; index =child) {//获取左儿子的位置
child = 2 *index;//如果只有左儿子
if (child ==n) {
child= 2 *index;
}//如果右儿子比左儿子的数值大
else if ((Integer) heap.get(child) < (Integer) heap.get(child + 1)) {
child++;
}//如果数值最大的儿子比temp的值大
if ((Integer) heap.get(child) >temp) {//交换堆中的child,和index位置的值
swap(heap, child, index);
}else{break;
}
}
}//打印链表
public static void print(Listlist) {for (int i = 1; i < list.size(); i++) {
System.out.print(list.get(i)+ " ");
}
System.out.println();
}//把堆中的a,b位置的值互换
public static void swap(List heap, int a, intb) {//临时存储child位置的值
int temp =(Integer) heap.get(a);//把index的值赋给child的位置
heap.set(a, heap.get(b));//把原来的child位置的数值赋值给index位置
heap.set(b, temp);
}//向最大堆中插入元素
public static void insert(List heap, intvalue) {//在数组的尾部添加要插入的元素
if (heap.size() == 0)
heap.add(0);//数组下标为0的位置不放元素
heap.add(value);//开始上升操作//heapUp2(heap, heap.size() - 1);
heapUp(heap, heap.size() - 1);
}//节点上浮,让插入的数和父节点的数值比较,当大于父节点的时候就和节点的值相交换
public static void heapUp(List heap, intindex) {//注意由于数值是从小标为一开始,当index = 1的时候,已经是根节点了
if (index > 1) {//保存父亲的节点
int parent = index / 2;//获取相应位置的数值
int parentValue =(Integer) heap.get(parent);int indexValue =(Integer) heap.get(index);//如果父亲节点比index的数值小,就交换二者的数值
if (parentValue
swap(heap, parent, index);//递归调用
heapUp(heap, parent);
}
}
}//非递归实现
public static void heapUp2(List heap, intindex) {int parent = 0;for (; index > 1; index /= 2) {//获取index的父节点的下标
parent = index / 2;//获得父节点的值
int parentValue =(Integer) heap.get(parent);//获得index位置的值
int indexValue =(Integer) heap.get(index);//如果小于就交换
if (parentValue
swap(heap, parent, index);
}
}
}/*根据树的性质建堆,树节点前一半一定是分支节点,即有孩子的,所以我们从这里开始调整出初始堆*/
public static void adjust(Listheap) {for (int i = heap.size() / 2; i > 0; i--)
adjust(heap, i, heap.size()- 1);
System.out.println("调整后的初始堆:");
print(heap);
}/*** 调整堆,使其满足堆得定义
*
*@parami
*@paramn*/
public static void adjust(List heap, int i, intn) {intchild;for (; i <= n / 2;) {
child= i * 2;if (child + 1 <= n && heap.get(child) < heap.get(child + 1))
child+= 1;/*使child指向值较大的孩子*/
if (heap.get(i)
swap(heap, i, child);/*交换后,以child为根的子树不一定满足堆定义,所以从child处开始调整*/i=child;
}else
break;
}
}//对一个最大堆heap排序
public static void heapSort(Listheap) {for (int i = heap.size() - 1; i > 0; i--) {/*把根节点跟最后一个元素交换位置,调整剩下的n-1个节点,即可排好序*/swap(heap,1, i);
adjust(heap,1, i - 1);
}
}public static voidmain(String args[]) {
List array = new ArrayList(Arrays.asList(null, 1, 2,5, 10, 3, 7, 11, 15, 17, 20, 9, 15, 8, 16));
adjust(array);//调整使array成为最大堆
delete(array,8);//堆中删除下标是8的元素
System.out.println("删除后");
print(array);
insert(array,99);//堆中插入
print(array);
heapSort(array);//排序
System.out.println("将堆排序后:");
print(array);
System.out.println("--------------------------------------------------");
List array1 = new ArrayList();
insert(array1,0);
insert(array1,1);
insert(array1,2);
insert(array1,5);
insert(array1,10);
insert(array1,3);
insert(array1,7);
insert(array1,11);
insert(array1,15);
insert(array1,17);
insert(array1,20);
insert(array1,9);
insert(array1,15);
insert(array1,8);
insert(array1,16);
print(array1);
System.out.println("--------------------------------------------------");
array= new ArrayList(Arrays.asList(null, 45, 36, 18, 53, 72,30, 48, 93, 15, 35));
adjust(array);
insert(array,80);//堆中插入
print(array);
delete(array,2);//堆中删除80的元素
print(array);
delete(array,2);//堆中删除72的元素
print(array);
}
}