两个最大堆的交集

Question

我有两个最大堆（用数组表示），大小为 m 的 H1 和大小为 n 的 H2，其中 n>m。我必须创建第三个最大堆，其中的元素来自 H1 和 H2 的交集。

基本解决方案（扫描两个数组）需要 O(n*m) 时间，并且没有利用最大堆属性。

其他想法？

Answer 1

如果可以进行散列，则与散列集进行交集，然后堆化。这是 O(n)，带有通常的注意事项。

如果无法进行散列，请与 H1 上设置的树（两者中较小的一个）进行交集。这是 O(n log m)，这与通常的元素清晰度下限一样好。

Answer 2

给定两个堆，你可以在 O(M log M + N log N) 时间内计算元素的交集，并且结果是有序的。一个有序的数组已经是一个堆，所以不需要更多的时间。

Python-语法示例：

# Given arrays heap1, heap2.

intersection = []
while len(heap1) > 0 and len(heap2) > 0:
    if heap1[0] == heap2[0]:
        # Common element, part of the intersection.
        intersection.append(heap1[0])
        heap.heappop(heap1)
        heap.heappop(heap2)
    elif heap1[0] > heap2[0]:
        heap.heappop(heap1)
    else:
        heap.heappop(heap2)

# intersection now contains the common elements of heap1 and heap2,
# in max-to-min order, so it already meets the heap invariant.

两个最大堆的交集

Intersection of two maximum heaps

algorithm

heap

set-intersection