从最大堆中的任何位置删除元素

时间:2015-09-24 00:41:34

标签: java max-heap

我有一个只有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内的堆中。我不确定这是否有帮助。

2 个答案:

答案 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);

    }