在二进制堆上实现迭代器

时间:2019-05-11 22:03:59

标签: iterator heap binary-heap

我正在寻找一种在二进制堆(最大或最小)上实现迭代器的方法。

也就是说,通过第i次使用其nextNode()函数,可以获得堆中第i个(更大或更小的)元素。

请注意,此操作不会真正提取堆的根!

我最初的想法是:

  • 实际上提取i元素,将它们压入堆栈,然后在获得第i个值后将它们重新插入堆中。每个函数调用都需要O(i * log(n))。
  • 保留一个辅助排序的数据结构,该结构可以允许在O(1)中查找下一个值,但是更新将占用O(n)。

我了解这些方法消除了使用堆的好处,因此我正在寻找一种更好的方法。

非常感谢。

1 个答案:

答案 0 :(得分:0)

目前尚不清楚用例是什么,因此很难说什么使解决方案可行或比其他解决方案更好。

也就是说,我建议对已经提出的一般“提取和排序”构想进行一些改动:如果我们可以很好地更改数据结构,则可以就地进行排序。

Wikipedia上建议的基本实现是引擎盖下的部分排序列表。我们可以(希望)一次性支付一次 O n log( n ))费用,以便在第一次{{ 1}}被调用,之后next() O (1)。至关重要的是,完全排序的列表仍然是有效的堆

此外,如果您考虑使用heapsort算法,则可以从第二阶段开始,因为您正在使用有效堆开始