While researching last week's article on Vector Search Trees, I found that two methods for constructing ball trees and the algorithm for querying ANNOY both involved *Priority Queues*. Since these are an important component of a number of different algorithms, it is worth examining them in detail.

Suppose we want to iterate over a set of items in a particular order. The naive way of doing this is to sort the list of items and then iterate over them. However, sorting is an expensive operation for large datasets, and we may want to add further items to the list while still iterating, which would necessitate re-sorting the list each time. We therefore need a more efficient way of tackling this.

Priority Queues address this by storing the data in a partially ordered data structure whose elements can be reordered efficiently when items are added or removed. Most implementations use a *heap*, which is a list of items with the following properties.

- The item at index \(i\) is the parent of the items at \(2i+1\) and \(2(i+1)\)
- The parent is less than or equal to each of its children.

These properties can be efficiently maintained by the following operations.

*Shift Up*- While an item is less than its parent (or not at the start of the list), swap it with its parent and check to see if its less than its new parent
*Shift Down*- While an item is greater than the smaller of its two children (or not at the end of the list), swap it with that child and check to see if it is smaller than either of its new children.

(*Note*: What I'm describing here is a *Min Heap*, which is used when we want to iterate over our items in ascending order. Most Python implementations of priority queues use this. There are also *Max Heaps*, which are used to iterate over items in descending order).

To add an item to the heap, we place it at the end, and then Shift Up until it reaches its proper place. When we remove the first item from the heap during iteration, we more the last item from the heap to the first position, and then Shift Down until it reaches its proper place.

There are several implementations of priority queues in Python heapq in the standard library, heapdict which implements a dictionary interface and allows the priority of items to be altered, and PriorityQueue in the standard Queue library, which is useful for sheduling data items to be processed by workers in a multithreaded application.

Prioritising tasks is an important part of many algorithms, so this is a useful tool to be aware of when designing an algorithm.

Previous | Next |
---|---|

Vector Search Trees | Graph Search Algorithms |