欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

java实现堆的操作方法(建堆,插入,删除)

程序员文章站 2023-12-04 11:37:34
如下所示: import java.util.arrays; //小顶堆的代码实现 public class heap { // 向下调整,顶端的...

如下所示:

import java.util.arrays;

//小顶堆的代码实现

public class heap {
	// 向下调整,顶端的大值往下调,主要用于删除和建堆,i表示要调整的节点索引,n表示堆的最有一个元素索引
	// 删除时候,i是0,建堆时候i从最后一个节点的父节点依次往前调整
	public static void fixdown(int[] data, int i, int n) {
		int num = data[i];
		int son = i * 2 + 1;
		while (son <= n) {
			if (son + 1 <= n && data[son + 1] < data[son])
				son++;
			if (num < data[son])
				break;
			data[i] = data[son];
			i = son;
			son = i * 2 + 1;
		}
		data[i] = num;
	}

	// 向上调整,小值往上走,用于增加,往上调整不需要制定最上面的索引,肯定是0
	public static void fixup(int[] data, int n) {
		int num = data[n];
		int father = (n - 1) / 2;
		// data[father] > num是进入循环的基本条件,father减到0就不会减少了
		// 当n等于0时,father=0;进入死循环,所以当n==0时,需要跳出循环
		while (data[father] > num && n != 0) {
			data[n] = data[father];
			n = father;
			father = (n - 1) / 2;
		}
		data[n] = num;
	}

	// 删除,n表示堆的最后一个元素的索引
	public static void delete(int[] data, int n) {
		data[0] = data[n];
		data[n] = -1;
		fixdown(data, 0, n - 1);
	}

	// 增加,i表示要增加的数字,n表示增加位置的索引,是堆的最后一个元素
	public static void insert(int[] data, int num, int n) {
		data[n] = num;
		fixup(data, n);
	}

	// 建堆,n表示要建堆的最后一个元素的索引
	public static void creat(int[] data, int n) {
		for (int i = (n - 1) / 2; i >= 0; i--)
			fixdown(data, i, n);
	}

	public static void main(string[] args) {
		int[] data = { 15, 13, 1, 5, 20, 12, 8, 9, 11 };
		// 测试建堆
		creat(data, data.length - 1);
		system.out.println(arrays.tostring(data));
		// 测试删除
		delete(data, data.length - 1);
		delete(data, data.length - 2);
		system.out.println(arrays.tostring(data));
		// 测试插入
		insert(data, 3, data.length - 2);
		system.out.println(arrays.tostring(data));

	}

}

以上这篇java实现堆的操作方法(建堆,插入,删除)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。