使用 ArrayList 进行堆放

Question

我编写了这个方法来测试我的优先级队列，它有一个 ArrayList 和一个 Comparator 作为参数：

@Test
  public void testPriorityQueue_ExtractMax() {
    PriorityQueue queue = new PriorityQueue(new IntegerComparator());
    Integer[] arrayExp={14, 9};
    queue.insert(16);
    queue.insert(9);
    queue.insert(14);
    queue.extractMax();
    assertArrayEquals(arrayExp, queue.getArray().toArray());
  }

现在，如果我执行它，它会说我的结果中的第一个元素是 9，但我必须是 14（我的队列的新根）。这些是 extract max 和 heapify 方法。我该如何解决？

public void heapify(int index) {
    int largest = index;
    int leftIndex = 2 * index + 1;
    int rightIndex = 2 * index + 2;

    if (leftIndex < queue.size() && c.compare(queue.get(index), (queue.get(leftIndex))) < 0)
      largest = leftIndex;

    if (rightIndex < queue.size() && c.compare(queue.get(largest), queue.get(rightIndex)) < 0)
      largest = rightIndex;

    if (largest != index) {
      swap(index, largest);
      heapify(largest);
    }
  }

  public T extractMax() {
    if (queue.size() == 0) return null;

    T min = queue.get(0);
    queue.remove(0);
    queue.set(0, queue.get(queue.size() - 1));
    heapify(0);
    return min;
  }

这是整数比较器：

public class IntegerComparator implements Comparator<Integer>{
  @Override
  public int compare(Integer l1, Integer l2) {
    int result = l1.compareTo(l2);
    if(result != 0)
      return result;
    return -1;
   }
}

EDIT_2：这是我的插入方法，会不会是问题：

public void insert(T elem) {
    int i = queue.size();
    int parentIndex = (i - 1) / 2;
    while (i > 0 && c.compare(elem, queue.get(parentIndex)) == 1) {
      queue.set(i, queue.get(parentIndex));
      i = parentIndex;
      parentIndex = (i - 1) / 2;
    }
    queue.add(i, elem);
  }

Answer 1

作为建议，考虑在索引 0 处添加一个未使用的元素，以便更直观地访问每个节点的父/子。

示例：

考虑堆heap = [-1, 6, 4, 5, 2, 1, 4, 3]，其中根定义为heap[1]，堆的数据位于从index = 1 到堆大小的位置。

要访问index 的节点的子节点，可以直观地说左子节点定义为heap[2 * index]，右子节点定义为heap[2 * index + 1]。同样，要访问index 节点的父节点，可以使用int 截断来访问父节点：

int parentInd = (int)(index/2);
T parent = heap[parentInd];

解决方案：

正如 raul1ro 所指出的，您正在丢失索引 0 中您不打算删除的数据。

在extractMax():

T min = queue.get(0);
queue.remove(0);
queue.set(0, queue.get(queue.size() - 1));

应该是：

T min = queue.get(0); //Get min
T temp = queue.get(queue.size() - 1); //Get the last element in heap as temp
queue.remove(queue.size - 1); //Remove the last element
queue.set(0, temp); //Set the root to the temp value so that you can heapify

这将使您在 extractMax() 时只丢失 1 个元素

希望有帮助！

Answer 2

当您这样做时queue.remove(i); 会自动将索引 i 之后的所有元素向左移动一个位置。

https://www.tutorialspoint.com/java/util/arraylist_remove.htm

而使用queue.set(0, queue.get(queue.size() - 1));，您只需将队列中最后一个索引的值设置为索引 0 并从索引 0 中丢失值，但它仍保留在最后一个位置。

读后插入方法

在queue.set(i, queue.get(parentIndex));，如果i = queue.size() 在第一步中，则意味着队列中不存在索引。