如何使用哈希表在min-heap上实现O(1)删除

时间:2013-08-21 17:18:27

标签: c++ algorithm

在某处阅读以下声明:

  

可以使用额外的哈希表快速删除   最小堆。

问题>如何合并priority_queueunordered_map以便我可以实现上述想法?

#include <queue>
#include <unordered_map>
#include <iostream>
#include <list>
using namespace std;

struct Age
{
  Age(int age) : m_age(age) {}
  int m_age;  
};

// Hash function for Age
class HashAge {
  public:
   const size_t operator()(const Age &a) const {
     return hash<int>()(a.m_age);
   }
};

struct AgeGreater
{
  bool operator()(const Age& lhs, const Age& rhs) const {
    return lhs.m_age < rhs.m_age;
  }
};

int main()
{
  priority_queue<Age, list<Age>, AgeGreater> min_heap;          // doesn't work
  //priority_queue<Age, vector<Age>, AgeGreater> min_heap;

  // Is this the right way to do it?
  unordered_map<Age, list<Age>::iterator, HashAge > hashTable;     
}

问题&GT;我无法完成以下工作:

priority_queue<Age, list<Age>, AgeGreater> min_heap;          // doesn't work

我必须使用list作为容器b / c列表的迭代器不受插入/删除的影响(Iterator invalidation rules

1 个答案:

答案 0 :(得分:4)

您无法使用提供的priority_queue数据结构执行此操作:

在优先级队列中,您不知道元素的存储位置,因此很难在恒定时间内删除它们,因为您无法找到元素。但是,如果你维护一个散列表,其中包含存储在散列表中的优先级队列中每个元素的位置,那么你可以快速找到并删除一个项目,尽管我希望在最坏的情况下使用log(N)时间,而不是常数时间。 (我不记得你是否经常摊销。)

为此,您通常需要滚动自己的数据结构,因为每次在优先级队列中移动项目时都必须更新哈希表。

我有一些示例代码可以在这里执行此操作:

http://code.google.com/p/hog2/source/browse/trunk/algorithms/AStarOpenClosed.h

它基于较旧的编码风格,但它可以完成这项工作。

举例说明:

/**
 * Moves a node up the heap. Returns true if the node was moved, false otherwise.
 */
template<typename state, typename CmpKey, class dataStructure>
bool AStarOpenClosed<state, CmpKey, dataStructure>::HeapifyUp(unsigned int index)
{
        if (index == 0) return false;
        int parent = (index-1)/2;
        CmpKey compare;

        if (compare(elements[theHeap[parent]], elements[theHeap[index]]))
        {
                // Perform normal heap operations
                unsigned int tmp = theHeap[parent];
                theHeap[parent] = theHeap[index];
                theHeap[index] = tmp;
                // Update the element location in the hash table
                elements[theHeap[parent]].openLocation = parent;
                elements[theHeap[index]].openLocation = index;
                HeapifyUp(parent);
                return true;
        }
        return false;
}

if语句中,我们在堆上执行正常的heapify操作,然后更新哈希表(openLocation)中的位置以指向优先级队列中的当前位置。