Алгоритм Кнута-Морриса-Пратта

Алгоритм Кнута — Морриса — Пратта (англ. Knuth–Morris–Pratt algorithm) — алгоритм поиска подстроки в строке.

Описание алгоритма

Дана цепочка [math]T[/math] и образец [math]P[/math]. Требуется найти все позиции, начиная с которых [math]P[/math] входит в [math]T[/math].
Построим строку [math]S = P\#T[/math], где [math]\#[/math] — любой символ, не входящий в алфавит [math]P[/math] и [math]T[/math]. Посчитаем на ней префикс-функцию [math]\pi()[/math]. Благодаря разделительному символу [math]\#[/math], выполняется [math]\forall i: \pi(i) \le |P|[/math]. Заметим, что по определению префикс-функции при [math]i \gt |P|[/math] и [math]\pi(i) = |P|[/math] подстроки длины [math]P[/math], начинающиеся с позиций [math]0[/math] и [math]i - |P| + 1[/math], совпадают. Соберем все такие позиции [math]i - |P| + 1[/math] строки [math]S[/math], вычтем из каждой позиции [math]|P| + 1[/math], это и будет ответ. Другими словами, если в какой-то позиции [math]i, \pi(i)=|P|[/math], то в этой позиции начинается очередное вхождение образца в цепочку.

Псевдокод

Пусть [math]p = |P|[/math], [math]t = |T|[/math].

count = 0
for (i = 0 .. (t - 1))
   if ([math]\pi[/math](p + i + 1) == p)
      answer[count++] = i + 1 - p

Время работы

Префикс-функция от строки [math]S[/math] строится за [math]O(S) = O(P + T)[/math]. Проход цикла по строке [math]S[/math] содержит [math]O(T)[/math] итераций. Итого, время работы алгоритма оценивается как [math]O(P + T)[/math]. Если мы хотим произвести множественный поиск образцов в тексте, то необходимо построить префикс-функцию для каждого из образцов в отдельности, тогда, учитывая, что длина образца обычно много меньше, чем длина текста, то общее время работы оценивается как [math]O(mT)[/math], где [math]m[/math]— количество образцов.

Оценка по памяти

Предложенная реализация имеет оценку по памяти [math]O(P+T)[/math]. Оценки [math]O(T)[/math] можно добиться за счет незапоминания значений [math]\pi()[/math] для позиций в [math]S[/math], меньших [math]p + 1[/math] (т.е. до начала цепочки [math]T[/math]).

Источники

• Википедия — Алгоритм Кнута — Морриса — Пратта
• Wikipedia — Knuth–Morris–Pratt algorithm
• Кормен, Т., Лейзерсон, Ч., Ривест, Р., Штайн — Алгоритмы: построение и анализ / пер. с англ. — изд. 2-е — М.: Издательский дом «Вильямс», 2009. — с.1036. — ISBN 978-5-8459-0857-5.