Очередь с приоритетами

Материал из Олимпиадное программирование в УлГТУ
Перейти к: навигация, поиск

Общие сведения

Очередь с приоритетами (англ. priority queue) — абстрактный контейнер, похожий на обычную очередь, но имеющий ряд особенностей:

  • Каждому элементу очереди с приоритетами сопоставлено некоторое значение, именуемое приоритетом этого элемента. Приоритеты допускают сравнение друг с другом;
  • Функция извлечения из очереди с приоритетами возвращает тот элемент, приоритет которого является максимальным.

Учащийся, выполняющий домашнее задание по нескольким предметам, может вводить различные показатели, определяющие порядок выполнения: от более любимых предметов к менее любимым, от менее сложных к более сложным, и так далее. Пожарная бригада планирует выезды сообразно с категорией сложности возгорания. Операционная система компьютера выделяет аппаратные ресурсы в первую очередь приложениям, запущенным с приоритетом реального времени, и лишь затем — остальным приложениям. Эти примеры позволяют пояснить назначение и принципы функционирования очередей с приоритетами.

Интерфейс

Интерфейс очереди с приоритетами в целом аналогичен интерфейсу обычной очереди, но в операции добавления появляется второй аргумент, а операция удаления возвращает элемент с наибольшим приоритетом:

void enqueue(T1 value, T2 priority) — вставка элемента value с приоритетом priority в очередь;
T1 dequeue() — извлечение элемента с максимальным приоритетом из очереди;
bool isEmpty() — проверка очереди на отсутствие элементов.

Демонстрация работы

В демонстрации показана очередь, в которой наивысший приоритет имеет наименьший элемент. Эта очередь построена на невозрастающей пирамиде (англ. min-oriented heap).
В статье рассматривается реализация очереди, в которой наивысший приоритет имеет наибольший элемент; соответствующая ей пирамида называется неубывающей (англ. max-oriented).

Реализация

Очередь с приоритетами, построенная на обычном массиве или списке, позволяет выполнять вставку за O(1), но поиск и удаление элемента с максимальным приоритетом будут иметь сложность O(N). Существует более быстрая реализация очереди с приоритетами, подразумевающая применение особого способа организации данных, получившего название пирамиды (англ. heap, также используется термин «куча»; не следует путать с динамическим хранилищем данных в памяти, которое также именуется кучей).

Определение пирамиды

Пирамида — это двоичное дерево, для элементов которого выполняется дополнительное условие (называемое основным свойством пирамиды): значение элемента в родительском узле больше (более точно, не меньше) значений во всех узлах-потомках.

Пирамида, отображаемая на массив

Пирамиду можно построить на массиве, используя соотношения между индексами ячеек массива для определения связей элементов. На иллюстрации показан пример пирамиды, содержащей 8 элементов, и её отображение на массив.

Если индексация элементов массива начинается с единицы, то непосредственные потомки элемента с индексом i будут иметь индексы (2 * i) и (2 * i + 1), а родителем будет являться элемент с индексом (i / 2). Так, потомками элемента с индексом 1 будут элементы с индексами 2 и 3, потомками элемента с индексом 2 — элементы с индексами 4 и 5, родителем элемента с индексом 8 будет элемент с индексом 4.

Если индексация элементов массива начинается с нуля, то непосредственные потомки элемента с индексом i будут иметь индексы (2 * i + 1) и (2 * i + 2), а родителем будет являться элемент с индексом ((i - 1) / 2). Так, потомками элемента с индексом 0 будут элементы с индексами 1 и 2, потомками элемента с индексом 1 — элементы с индексами 3 и 4, родителем элемента с индексом 7 будет элемент с индексом 3.

В целях упрощения дальнейшего кода определим вспомогательные функции parent(), leftChild() и rightChild():

int parent(int i) {
    return (i - 1) / 2;
}

int leftChild(int i) {
    return 2 * i + 1;
}

int rightChild(int i) {
    return 2 * i + 2;
}

Пирамида в данной реализации является почти полным двоичным деревом: только самый нижний уровень может быть не заполнен полностью, и его заполнение происходит строго слева направо. Можно показать, что в N-элементной пирамиде ⌈N/2⌉ последних элементов являются листьями, то есть не имеют потомков. При индексации массива с единицы листья имеют индексы от (N / 2 + 1) до N, при индексации с нуля — от (N / 2) до (N - 1).

Высотой пирамиды называется количество рёбер на самом длинном пути от вершины пирамиды до какого-либо листа. Достаточно легко убедиться, что высота почти полной пирамиды из N элементов равна ⌊log2N⌋.

Поддержка основного свойства пирамиды

Добавление, изменение или удаление некоторого элемента пирамиды может нарушить основное свойство пирамиды. Каждое такое нарушение можно отнести к одному из двух основных типов:

  • Значение элемента больше, чем у его родителя, и поэтому элемент должен занять место выше, чем имеет сейчас;
  • Значение элемента меньше, чем у хотя бы одного из его потомков, и поэтому элемент должен занять место ниже, чем имеет сейчас.
Восстановление пирамиды — операция up()

Процедура up(), которая будет исправлять нарушения первого типа, обменивает заданный элемент с его родителем до тех пор, пока значение элемента не станет меньше значения родителя, либо элемент не окажется на вершине пирамиды. Процедура принимает индекс элемента, который вызывает нарушение основного свойства пирамиды. Приведённый ниже код работает для массива, в котором индексация начинается с нуля.

void up(int i) {
    while (i != 0 && a[i] > a[parent(i)]) {
        swap(a[i], a[parent(i)];
        i = parent(i);
    }
}
Восстановление пирамиды — операция down()

Процедура down(), исправляющая нарушения второго типа, обменивает заданный элемент с максимальным из его потомков до тех пор, пока значение элемента не станет больше значений потомков, либо элемент не окажется листом. Обмен происходит с максимальным из потомков, так как только в этом случае основное свойство пирамиды гарантированно поддерживается после обмена. Как и в предыдущем случае, процедура принимает индекс неверно расположенного элемента в массиве, а индексация элементов начинается с нуля. Общее количество элементов в очереди равно size.

void down(int i) {
    while (i < size / 2) {
        int maxI = leftChild(i);
        if (rightChild(i) < size && a[rightChild(i)] > a[leftChild(i)])
            maxI = rightChild(i);
        if (a[i] >= a[maxI])
            return;
        swap(a[i], a[maxI]);
        i = maxI;
    }
}

Построение очереди с приоритетами на пирамиде

Важная особенность пирамиды состоит в том, что максимальное из хранящихся в ней значений находится на её вершине. В совокупности с тем фактом, что показанные ранее операции восстановления пирамиды up() и down() производят количество перемещений, не превышающее высоты пирамиды, это позволяет разработать эффективную реализацию очередей с приоритетами.

Прежде всего, эта реализация потребует описания структуры элемента (так как элемент хранит не только приоритет, но и значение) и объявления массива, который будет организован как пирамида. Операции восстановления пирамиды должны будут сравнивать поля .priority. Для хранения количества элементов в очереди отводится отдельная переменная size, которая в конструкторе инициализируется значением 0.

static const int MAX_SIZE = 100;
struct Elem {
    int val;
    int priority;
    Elem(int v = 0, int p = 0) {
        val = v;
        priority = p;
    }
} a[MAX_SIZE];
int size;

Добавление элемента происходит в ячейку a[size]. Так как после добавления может быть нарушено основное свойство пирамиды, требуется дополнительно вызвать процедуру up(). Общая сложность операции добавления составит O(logN).

void enqueue(int value, int priority) {
    if (size + 1 == MAX_SIZE)
        /* обработка ошибки - переполнение очереди */
    a[size++] = Elem(value, priority);
    up(size - 1);
}

При удалении элемента можно поступить следующим образом: переместить на вершину пирамиды её последний элемент и выполнить для него операцию down(). Сложность удаления равна O(logN).

int dequeue() {
    if (size == 0)
        /* обработка ошибки - нет элементов для извлечения */
    swap(a[0], a[--size]);
    down(0);
    return a[size];
}

Ниже приведён полный код реализации очереди с приоритетами. Переполнение и попытка извлечения из пустой очереди выявляются с помощью конструкции assert (заголовочный файл <assert.h> либо <cassert>).

class PriorityQueue {

    static const int MAX_SIZE = 100;
    struct Elem {
        int val;
        int priority;
        Elem(int v = 0, int p = 0) {
            val = v;
            priority = p;
        }
    } a[MAX_SIZE];
    int size;

    void up(int i) {
        while (i != 0 && a[i].priority > a[(i - 1) / 2].priority) {
            swap(a[i], a[(i - 1) / 2]);
            i = (i - 1) / 2;
        }
    }

    void down(int i) {
        while (i < size / 2) {
            int maxI = 2 * i + 1;
            if (2 * i + 2 < size && a[2 * i + 2].priority > a[2 * i + 1].priority)
                maxI = 2 * i + 2;
            if (a[i].priority >= a[maxI].priority)
                return;
            swap(a[i], a[maxI]);
            i = maxI;
        }
    }

public:

    PriorityQueue() {
        size = 0;
    }

    void enqueue(int value, int priority) {
        assert(size + 1 < MAX_SIZE);
        a[size++] = Elem(value, priority);
        up(size - 1);
    }

    int dequeue() {
        assert(size > 0);
        swap(a[0], a[--size]);
        down(0);
        return a[size];
    }

    bool isEmpty() {
        return size == 0;
    }

};

Очередь с приоритетами в STL

В стандартной библиотеке шаблонов C++ присутствует шаблон priority_queue<T>. Для возможности его использования требуется подключить заголовочный файл <queue> и пространство имён std.

#include <iostream>
#include <queue>
using namespace std;
int main() {
    priority_queue<int> pq;
    q.push(1);
    q.push(3);
    q.push(2);
    while (!s.empty()) {
        cout << q.top() << ' ';
        q.pop();
    }            
    return 0;       //результат "3 2 1 "
}

Упорядочение элементов в очереди производится по убыванию; в качестве операции сравнения по умолчанию используется оператор <. Для применения другой функции сравнения требуется передать её третьим параметром в конструкторе очереди (второй параметр — тип базового контейнера, по умолчанию vector). Например, определение std::priority_queue< int, std::vector<int>, std::greater<int> > позволяет получить очередь, в которой наивысшим считается минимальный приоритет (функциональный объект greater объявлен в заголовочном файле <functional>).

Набор методов для очереди с приоритетами в STL почти полностью аналогичен таковому для обычной очереди:

priority_queue<T>() — конструктор;
void push(const T& x) — добавление элемента в очередь. Приоритет элемента определяется его значением;
void pop() — удаление элемента с максимальным приоритетом. Обратите внимание на то, что значение удаляемого элемента не возвращается;
T& top() — получение значения элемента с максимальным приоритетом. Этот метод не производит удаление элемента;
bool empty() — проверка очереди с приоритетами на пустоту;
size_t size() — получение количества элементов в очереди с приоритетами. Метод возвращает беззнаковое целое число.

Очередь с приоритетами в STL создаётся на базе шаблона последовательного контейнера с произвольным доступом (vector или deque), и для поддержания внутренней структуры она использует процедуры работы с пирамидами, определённые в заголовочном файле <algorithm>. Каждая из них принимает пару итераторов произвольного доступа It, указывающих диапазон элементов (как правило, это begin() и end(), покрывающие все элементы в контейнере). По умолчанию для сравнения во всех процедурах используется оператор <; собственный функциональный объект можно передать третьим параметром.

void make_heap(It first, It last) — создание пирамиды из диапазона элементов [first, last);
void push_heap(It first, It last) — добавление элемента, находящегося перед last, в пирамиду [first, last-1), так что весь диапазон [first, last) становится пирамидой;
void pop_heap(It first, It last) — перемещение элемента с максимальным приоритетом (на который изначально указывает first) в конец и формирование пирамиды в диапазоне [first, last-1) из оставшихся элементов;
void sort_heap(It first, It last) — преобразование пирамиды [first, last) в упорядоченный интервал. После вызова процедуры диапазон перестаёт быть пирамидой.

Пирамидальная сортировка

Пирамидальная сортировка (англ. heapsort) — улучшенная версия сортировки выбором. Обычная сортировка выбором предполагает, что N раз из данного массива извлекается наименьший элемент и ставится на соответствующее место. Если для выбора наименьшего элемента используется последовательный поиск, то каждый раз просматривается весь массив, что даёт сложность каждой итерации O(N) и итоговую сложность O(N2)

Вместо последовательного поиска наименьшего элемента можно использовать пирамиду, позволяющую на каждом шаге извлекать минимальный элемент за O(1) и восстанавливаться за O(logN), что даст итоговую сложность O(NlogN). Чтобы не перемещать саму пирамиду, на каждом шаге можно извлекать максимальный элемент и перемещать его в конец. Построить пирамиду на непустом данном массиве можно двумя способами:

  • Проходя от начала массива до конца, последовательно выполнить для каждого элемента операцию up() (что в целом аналогично построению очереди с приоритетами за счёт выполнения операции enqueue() для каждого элемента массива). Сложность O(NlogN);
  • Построить пирамиду «снизу вверх», проходя от конца массива до начала и вызывая процедуру down() для всех элементов. Вызов down() для листьев не предусматривает никаких действий, поэтому можно начинать цикл с (size / 2 - 1). В нескольких источниках, в том числе в книге Т. Кормена «Алгоритмы: построение и анализ», показано, что сложность данного метода снижается до O(N).

Пусть функция down() использует значения a[] и size, переданные ей через параметры. Тогда код пирамидальной сортировки имеет следующий вид:

void heapsort(int a[], int size) {
    for (int i = size / 2 - 1; i >= 0; i--)
        down(a, size, i);
    for (int i = 0; i < size; i++) {
        swap(a[0], a[size - 1 - i];
        down(a, size - i, 0);
    }
}

Свойства пирамидальной сортировки:

  • Сложность O(NlogN);
  • Как и сортировка выбором, не требует дополнительной памяти, в отличие от, например, сортировки слиянием;
  • Как и сортировка выбором, не является стабильной (например, массив [1-a, 1-b], где числа являются ключами, после сортировки примет вид [1-b, 1-a]);
  • Как и сортировка выбором, не является адаптивной.

Демонстрация работы пирамидальной сортировки:

Ссылки

Теория:

Код:

Задачи: