Cache-oblivious distribution sort

Distribution sort operates on a contiguous array of ${\displaystyle N}$ elements. To sort the elements, it performs the following:

1. Partition the array into ${\displaystyle {\sqrt {N}}}$ contiguous subarrays of size ${\displaystyle {\sqrt {N}}}$, and recursively sort each subarray.
2. Distribute the elements of the sorted subarrays into ${\displaystyle q\leq {\sqrt {N}}}$ buckets ${\displaystyle B_{1},B_{2},\ldots ,B_{q}}$ each of size at most ${\displaystyle 2{\sqrt {N}}}$ such that for every i from 1 to q-1, every element of bucket ${\displaystyle B_{i}}$ is not larger than any element in ${\displaystyle B_{i+1}.}$ This distribution step is the main step of this algorithm, and is covered in more detail below.
3. Recursively sort each bucket.
4. Output the concatenation of the buckets.

As mentioned in step 2 above, the goal of the distribution step is to distribute the sorted subarrays into q buckets ${\displaystyle B_{1},B_{2},\ldots ,B_{q}.}$ The distribution step algorithm maintains two invariants. The first is that each bucket has size at most ${\displaystyle 2{\sqrt {N}}}$ at any time, and any element in bucket ${\displaystyle B_{i}}$ is no larger than any element in bucket ${\displaystyle B_{i+1}.}$ The second is that every bucket has an associated pivot, a value which is greater than all elements in the bucket.

Initially, the algorithm starts with one empty bucket with pivot ${\displaystyle \infty }$. As it fills buckets, it creates new buckets by splitting a bucket into two when it would be made overfull (by having at least ${\displaystyle (2{\sqrt {N}}+1)}$ elements placed into it). The split is done by performing the linear time median finding algorithm, and partitioning based on this median. The pivot of the lower bucket will be set to the median found, and the pivot of the higher bucket will be set to the same as the bucket before the split. At the end of the distribution step, all elements are in the buckets, and the two invariants will still hold.

To accomplish this, each subarray and bucket will have a state associated with it. The state of a subarray consists of an index next of the next element to be read from the subarray, and a bucket number bnum indicating which bucket index the element should be copied to. By convention, ${\displaystyle bnum=\infty }$ if all elements in the subarray have been distributed. (Note that when we split a bucket, we have to increment all bnum values of all subarrays whose bnum value is greater than the index of the bucket that is split.) The state of a bucket consists of the value of the bucket's pivot, and the number of elements currently in the bucket.

Consider the follow basic strategy: iterate through each subarray, attempting to copy over its element at position next. If the element is smaller than the pivot of bucket bnum, then place it in that bucket, possibly incurring a bucket split. Otherwise, increment bnum until a bucket whose pivot is large enough is found. Though this correctly distributes all elements, it does not exhibit a good cache performance.

Instead, the distribution step is performed in a recursive divide-and-conquer. The step will be performed as a call to the function distribute, which takes three parameters i, j, and m. distribute(i, j, m) will distribute elements from the i-th through (i+m-1)-th subarrays into buckets, starting from ${\displaystyle B_{j}}$. It requires as a precondition that each subarray r in the range ${\displaystyle i,\ldots ,i+m-1}$ has its ${\displaystyle bnum[r]\geq j}$. The execution of distribute(i, j, m) will guarantee that each ${\displaystyle bnum[r]\geq j+m}$. The whole distribution step is distribute${\displaystyle (1,1,{\sqrt {N}})}$. Pseudocode for the implementation of distribute is shown below:

def distribute(i, j, m: int) -> None:
    """Distribute through recursive divide-and-conquer."""
    if m == 1:
        copy_elems(i, j)
    else:
        distribute(i, j, m / 2)
        distribute(i + m / 2, j, m / 2)
        distribute(i, j + m / 2, m / 2)
        distribute(i + m / 2, j + m / 2, m / 2)

The base case, where m=1, has a call to the subroutine copy_elems. In this base case, all elements from subarray i that belong to bucket j are added at once. If this leads to bucket j having too many elements, it splits the bucket with the procedure described beforehand.