我有一个只有x和y的Point对象,我有一个看起来像这样的Heap数据结构:
class MaxHeap{
public Point[] heap;
public int size;
public int maxsize;
public MaxHeap(int maxsize){
this.maxsize = maxsize;
this.size = 0;
heap = new Point[this.maxsize+1];
heap[0] = new Point(-1,-1); //Heap is empty
}
public int parent(int pos){
return pos /2;
}
public int leftChild(int pos){
return (2 * pos);
}
public int rightChild(int pos){
return (2 * pos) +1;
}
public boolean isLeaf(int pos){
if (pos >= (size / 2) && pos <= size){
return true;
}
return false;
}
public void swap (int fpos, int spos){
Point tmp;
tmp = heap[fpos];
heap[fpos] = heap[spos];
heap[spos] = tmp;
}
public void maxHeapify(int pos){
if (!isLeaf(pos)){
if (heap[pos].getY() < heap[leftChild(pos)].getY() || heap[pos].getY() < heap[rightChild(pos)].getY()){
swap(pos, leftChild(pos));
maxHeapify(leftChild(pos));
}
else{
swap(pos, rightChild(pos));
maxHeapify(rightChild(pos));
}
}
}
public void insert (Point p){
heap[++size] = p;
int current = size;
while (heap[current].getY() > heap[parent(current)].getY()){
swap(current, parent(current));
current = parent(current);
}
}
我正在尝试实现一种方法来从堆中删除任何Point,而不是传统的删除它只是删除顶部。我不完全确定如何去做这件事。我以为我可以将Point的索引存储在Point内的堆中。我不确定这是否有帮助。
答案 0 :(得分:1)
就是这样,您是否意识到Java中存在标准的堆实现,称为PriorityQueue?您可以将其用作removeAt(int i)
实施方式的参考。
回到你的问题。
为了从队列中删除中间元素,您需要将其替换为队列的最后一个元素(将队列缩小一个元素)并尝试&#34; heapify&#34;这个元素下来了。如果元素仍然存在(两个孩子都比元素大),你需要&#34;堆积&#34;它了。
关于你问题的第二部分。我不建议在Point
类中存储队列索引,从而使得队列可识别。更好的方法是将Map从点保持到队列内的索引(此映射可以由IdentityHashMap[Point, Integer]
表示)。当您在队列中进行更改时,不要忘记在此地图中进行适当的更改,例如插入,删除元素,交换元素等等。
答案 1 :(得分:0)
这是一个回答:
public void removeSpecificElement(int i) {
heap[i] = heap[size];
size--;
while (getParent(i) < heap[i] && i > 1 ) {
swapElements(heap[i], getParent(i));
i = getParent(i);
}
heapifyUp(i);
}