实现自定义小根堆前我们首先要了解两个公式
1.父节点到子节点:2N+1
2.子节点到父节点:(N-1)/2
添加元素时,只需通过添加的元素与其父节点元素的大小进行比较,若子节点元素大于父节点则位置调换,这里便需要用到公式2
删除原始时,直接将最后一位元素赋值给第一位,这样做的好处便是可以尽可能少的破坏原来已经形成的小根堆,这里需要用到公式1,要注意的时在父节点与子节点比较时,首先要将两个字节点的值进行比较,取小的与父节点进行比较,这样才能保证交换后是一个小根堆
public class HeapNumber {
@Override
public String toString() {
return Arrays.toString(data);
}
private int[] data;
private int size;
public HeapNumber() {
this(10);
}
public HeapNumber(int size) {
this.data = new int[size];
this.size = 0;
}
public void add(int value) {
if (isFull()) {
data = Arrays.copyOf(data ,data.length * 2);
}
if (size == 0) {
data[this.size] = value;
this.size++;
} else {
data[this.size] = value;
this.size++;
up();
}
}
public void up() {
for (int i = this.size-1; i > 0; i = ( i - 1 ) / 2) {
if (data[i] < data[( i - 1 ) / 2]) {
int tmp = data[i];
data[i] = data[( i - 1 ) / 2];
data[( i - 1 ) / 2] = tmp;
}
}
}
public boolean isFull() {
return this.size == data.length;
}
public int remove() {
int value = data[0];
data[0] = data[size-1];
data[size-1] = 0;
size--;
down();
return value;
}
public boolean down() {
int start = 0;
for(int i = 2*start+1; i < size-1;i = 2*i+1) {
int tmp = data[start];
if(data[i] > data[i+1]) {
i= i+1;
}
if(data[i] < tmp) {
data[start] = data[i];
start = i;
} else {
break;
}
data[i] = tmp;
}
return true;
}
public void show() {
for(int i = 0;i < data.length;i++) {
System.out.println(data[i]);
}
}
public int gettop() {
return data[0];
}