Meet-in-the-middle: оптимизация перебора и не только

В прошлой статье ^[1] я писал про эвристические методы оптимизации перебора. В этой статье я расскажу вам о ещё одной, но уже асимптотической оптимизации — meet-in-the-middle.

Типичные для этого метода снижения асимптотики: и .

Вступление

Метод заключается в том, чтобы разделить задачу пополам, получить какие-то данные и сопоставить их друг с другом.

Meet-in-the-middle имеет смысл применять, если для конкретной задачи выполняются два условия:
1) Время обработки половины данных асимптотически меньше времени получения итогового ответа.
2) Известен асимптотически более быстрый способ получения ответа для всей задачи с использованием информации об обработки её половинок.

Чтобы появилось понимание, лучше всего посмотреть на примеры (даны в порядке увеличения предполагаемой сложности):

Пример №1: Оптимизация перебора

Дано N чисел. Надо выбрать k чисел так, чтобы их сумма была равна 0 (одинаковые элементы могут быть использованы несколько раз).

Наивное решение:
Перебрать все вариантов выбора k чисел, пересчитывая сумму по ходу рекурсии. Асимптотика: .

Решение с meet-in-the-middle для чётного k:
1) Мысленно разобьём k чисел, которые надо найти, на две кучки по k/2.
2.1) Выпишем все сумм.
2.2) Отсортируем массив сумм.
2.3) Для каждой суммы s попытаемся найти бинарным поиском -s.
2.4) Если нашли, то вот он, наш ответ. Если нет, то такой комбинации не существует.
3) Profit: .
Изменить этот алгоритм для нечётного k, думаю, не составит труда для вас.

Если N = 30, а k = 12, то meet-in-the-middle будет работать примерно минуту, а наивный алгоритм — 17 лет.

Пример №2: Оптимизация поиска кратчайшего пути в огромном графе состояний

У фокусника есть колода из 36 игральных карт. Изначально они лежат в порядке 1, 2, 3..., а он хочет получить какую-то конкретную перестановку, причем не более, чем за k шагов. На каждом шаге фокусник перекладывает m подряд идущих карт в начало колоды.

Картинка для пояснения

Задача состоит в том, чтобы узнать может ли он получить нужную перестановку.

Определим граф состояний S, как граф, в котором вершины задаются текущей перестановкой карт, а рёбра возможностью перейти от одной перестановки к другой за 1 шаг. Стоит отметить, что в этом графе 36! вершин, но из каждой вершины выходит 36-m+1 ребро, что относительно немного.

Наивное решение:
Запустить поиск в ширину на графе состояний S. Если дошли до нужного состояния или до вершины удалённой на k+1, то завершить поиск. Асимптотика: .

Решение с meet-in-the-middle:
1) Запустим два поиска в ширину из начальной и конечной вершины с максимальной глубиной k/2. Запишем два множества: все состояния, в которых побывали поиски в ширину.
2) Если эти множества пересекаются, значит есть путь, если нет, то и пути нет.
3) Profit: .

Пример №3: Подсчёт количества «хороших» подмножеств

for mask in range(1 << N):
    dp[mask] = 1 + sum([dp[mask & matrix[i]] for i in range(N) if ((mask & (1 << i)) > 0)])

Заключение

Надеюсь, я смог доходчиво объяснить концепцию meet-in-the-middle. Если остались вопросы — задавайте в комментариях, я постараюсь ответить. Спасибо за внимание.

Автор: Frommi

Источник ^[3]