Сведение задачи LCA к задаче RMQ — различия между версиями

Материал из Викиконспекты
Перейти к: навигация, поиск
м (rollbackEdits.php mass rollback)
 
(не показано 40 промежуточных версий 9 участников)
Строка 1: Строка 1:
== Постановка задачи LCA ==
+
{{Шаблон:Задача
 +
|definition =  
 +
Пусть дано корневое дерево <tex>T</tex>. На вход подаются запросы вида <tex>(u,\;v)</tex>, для каждого запроса требуется найти их наименьшего общего предка.
 +
}}
 
{{Определение  
 
{{Определение  
|definition = '''Наименьшим общим предком (least common ancestor)''' двух узлов <tex>u, v</tex> в корневом дереве <tex>T</tex> называется узел <tex>w,</tex> который среди всех узлов, являющихся предками как узла <tex>u,</tex> так и <tex>v,</tex> имеет наибольшую глубину.
+
|id=lca_suf_tree
 +
|definition = '''Наименьшим общим предком''' (англ. ''least common ancestor'') двух узлов <tex>u</tex> и  <tex>v</tex> в корневом дереве <tex>T</tex> называется узел <tex>w</tex>, который среди всех узлов, являющихся предками как узла <tex>u</tex>, так и <tex>v</tex>, имеет наибольшую глубину.
 
}}
 
}}
Пусть дано корневое дерево <tex>T.</tex> На вход подаются запросы вида <tex>(u,\;v),</tex> для каждого запроса требуется найти их наименьшего общего предка.
 
  
 
== Алгоритм ==
 
== Алгоритм ==
 +
=== Идея ===
 +
Будем решать задачу <tex>LCA</tex>, уже умея решать задачу <tex>RMQ</tex>. Тогда поиск наименьшего общего предка <tex>i</tex>-того и <tex>j</tex>-того элементов сводится к запросу минимума на отрезке массива, который будет введен позднее.
 +
 
=== Препроцессинг ===
 
=== Препроцессинг ===
1) В каждом узле будет храниться глубина узла в корневом дереве <tex>T.</tex>
+
Для каждой вершины <tex>T</tex> определим глубину с помощью следующей рекурсивной формулы:
:<tex>depth(u)= \begin{cases}
+
:<tex>\mathrm{depth}(u) = \begin{cases}
0 & u = root(T),\\
+
0 & u = \mathrm{root}(T),\\
depth(v) + 1 & u = son(v).
+
\mathrm{depth}(v) + 1 & u = \mathrm{son}(v).
 
\end{cases}</tex>
 
\end{cases}</tex>
 +
Ясно, что глубина вершины элементарным образом поддерживается во время обхода в глубину.
 +
 +
Запустим [[Обход в глубину, цвета вершин|обход в глубину]] из корня, который будет вычислять значения следующих величин:
 +
#Cписок глубин посещенных вершин <tex>d</tex>. Глубина текущей вершины добавляется в конец списка при входе в данную вершину, а также после каждого возвращения из её сына.
 +
#Список посещений узлов <tex>\mathtt{vtx}</tex>, строящийся аналогично предыдущему, только добавляется не глубина а сама вершина.
 +
#Значение функции <tex>\mathtt{I}[u]</tex>, возвращающей индекс в списке глубин <tex>d</tex>, по которому была записана глубина вершины <tex>u</tex> (например на момент входа в вершину).
 +
 +
Вот таким образом будут выглядеть эти три массива после обхода в глубину:
  
2) Запустим [[Обход в глубину, цвета вершин|обход в глубину]] из корня, который будет строить список посещений узлов. Глубина текущей вершины добавляется в список при входе в эту вершину, а также после каждого возвращения из её сына.
+
[[Файл:HoD8KSiOzTg.jpg‎ | мини | left | 700x500px| Пример массива <tex>\mathtt{vtx}</tex>]]
 +
<br clear="all">
  
 
=== Запрос ===
 
=== Запрос ===
Обозначим <tex>I[u]</tex> {{---}} функция, возвращающая любой индекс ячейки в списке глубин <tex>depth</tex>, в котором хранится глубина узла <tex>u</tex>. Ее можно строить во время обхода в глубину.
+
Будем считать, что <tex>\mathrm{rmq}(d,l,r)</tex> возвращает индекс минимального элемента в <tex>d</tex> на отрезке <tex>[l..r]</tex>. Тогда ответом на запрос <tex>\mathrm{lca}(u, v)</tex>, где <tex>\mathtt{I}[u] \leqslant \mathtt{I}[v]</tex>, будет <tex>\mathtt{vtx}[\mathrm{rmq}(d,\mathtt{I}[u], \mathtt{I}[v])]</tex>.
Пусть имеется запрос пара узлов <tex>u, v.</tex> В результате обхода в глубину получился список глубин вершин, в котором наименьшему общему предку вершин <tex>u, v</tex> соответствует минимальная глубина на отрезке <tex>depth[I[u], I[v]]. </tex> Для определённости <tex>I[u] \le I[v].</tex>
 
 
 
 
=== Доказательство корректности алгоритма ===
 
=== Доказательство корректности алгоритма ===
 
{{Теорема
 
{{Теорема
 
|statement=
 
|statement=
Приведенный выше алгоритм работает верно.
+
Наименьшему общему предку вершин <tex>u, v</tex> соответствует минимальная глубина на отрезке <tex>d[\mathtt{I}[u], \mathtt{I}[v]]</tex>.
 
|proof=
 
|proof=
Рассмотрим два узла <tex>u, v</tex> корневого дерева <tex>T</tex>. Для определенности считаем, что <tex>u</tex> является первой при поиске в глубину. На отрезке <tex>depth[I[u]..end]</tex> (<tex>end</tex> {{---}} последний элемент в <tex>depth</tex>) хранятся узлы посещенные после <tex>u</tex> и, быть может, некоторые вершины из поддерева с корнем <tex>u</tex>(которые имеют глубину больше глубины <tex>u</tex>). Аналогично на <tex>depth[1..I[v]]</tex> {{---}} вершины, посещенные до <tex>v</tex> и некоторые вершины из поддерева <tex>v</tex>. Рассмотрим теперь отрезок <tex>depth[I[u]..I[v]]</tex>. Поскольку этот отрезок {{---}} путь из <tex>u</tex> в <tex>v</tex>, он проходит через их наименьшего общего предка <tex>w</tex>(в дереве есть только один простой путь между вершинами). Покажем, что его глубина минимальна на отрезке <tex>depth[I[u]..I[v]]</tex>. Допустим обратное. Все потомки <tex>w</tex> имеют глубину больше. Но тогда получим, что поиск в глубину вышел из поддерева вершины <tex>w</tex> раньше, чем посетил вершину <tex>v</tex>.
+
Рассмотрим два узла <tex>u, v</tex> корневого дерева <tex>T</tex>. Рассмотрим отрезок <tex>d[\mathtt{I}[u]..\mathtt{I}[v]]</tex>. Поскольку этот отрезок {{---}} путь из <tex>u</tex> в <tex>v</tex>, он проходит через их наименьшего общего предка <tex>w</tex> (в дереве есть только один простой путь между вершинами), а следовательно минимум на отрезке никак не больше глубины <tex>w</tex>. Заметим, что в момент добавления <tex>\mathtt{I}[u]</tex> обход посещал поддерево с корнем <tex>w</tex>. В момент добавления <tex>\mathtt{I}[v]</tex> мы все еще в поддереве с корнем <tex>w</tex>. Значит, и на отрезке между <tex>\mathtt{I}[u]</tex> и <tex>\mathtt{I}[v]</tex> мы находились внутри поддерева с корнем <tex>w</tex>. Отсюда сделаем заключение, что на рассматриваемом отрезке не посещалась вершина, отличная от <tex>w</tex>, с глубиной меньшей либо равной глубины <tex>w</tex>, т. к. подобной вершины нет в поддереве с корнем <tex>w</tex>.
}}
+
}}.
  
 
== Пример ==
 
== Пример ==
 
Рассмотрим дерево на рисунке 1. Найдем наименьшего общего предка вершин, помеченных красным цветом.
 
Рассмотрим дерево на рисунке 1. Найдем наименьшего общего предка вершин, помеченных красным цветом.
Список глубин, получающийся в результате обхода в глубину - <tex>[0, 1, 2, 1, 2, 1, 0, 1, 0].</tex>
+
Список глубин, получающийся в результате обхода в глубину {{---}} <tex>[0, 1, 2, 1, 2, 1, 0, 1, 0].</tex>
 
Глубина наименьшего общего предка красных вершин равна минимуму на отрезке <tex>[2, 1, 0, 1].</tex>
 
Глубина наименьшего общего предка красных вершин равна минимуму на отрезке <tex>[2, 1, 0, 1].</tex>
[[Файл:Lca to rmq.png|thumb|left|рис. 1]]
+
[[Файл:Lca to rmq.png(2).png|thumb|center|400x200px|Рисунок к примеру]]
 
<div style='clear:left;'></div>
 
<div style='clear:left;'></div>
  
 
== Сложность ==
 
== Сложность ==
  
Для нахождения минимального элемента на отрезке можно использовать [[Дерево отрезков. Построение|дерево отрезков]]. Длина массива глубин будет равна <tex>(2n - 1),</tex> т.е. дерево отрезков будет построено за <tex>O(n).</tex> Таким образом, препроцессинг работает за <tex>O(n).</tex> Время выполнения запроса равно времени запроса минимального элемента на отрезке в дереве отрезков, т.е. <tex>O(\log n).</tex>
+
Для нахождения минимального элемента на отрезке можно использовать [[Алгоритм Фарака-Колтона и Бендера|алгоритм Фарака-Колтона и Бендера]] для <tex>\pm 1RMQ</tex>, т. к. соседние элементы в списке глубин отличаются не более чем на единицу. Длина списка глубин составляет <tex>(2n - 1)</tex>, таким образом, препроцессинг работает за <tex>O(n)</tex>. Время выполнения запроса равно времени запроса минимального элемента на отрезке {{---}} <tex>O(1)</tex>.
  
 
== См.также ==
 
== См.также ==
Строка 43: Строка 56:
 
*[[Алгоритм Фарака-Колтона и Бендера]]
 
*[[Алгоритм Фарака-Колтона и Бендера]]
 
*[[Сведение задачи RMQ к задаче LCA]]
 
*[[Сведение задачи RMQ к задаче LCA]]
== Ссылки ==
+
== Источники информации ==
 
*[http://e-maxx.ru/algo/lca Наименьший общий предок. Нахождение за O (sqrt (N)) и O (log N) с препроцессингом O (N)]
 
*[http://e-maxx.ru/algo/lca Наименьший общий предок. Нахождение за O (sqrt (N)) и O (log N) с препроцессингом O (N)]
  
 
[[Категория: Алгоритмы и структуры данных]]
 
[[Категория: Алгоритмы и структуры данных]]
 
[[Категория: Задача о наименьшем общем предке]]
 
[[Категория: Задача о наименьшем общем предке]]

Текущая версия на 19:21, 4 сентября 2022

Задача:
Пусть дано корневое дерево [math]T[/math]. На вход подаются запросы вида [math](u,\;v)[/math], для каждого запроса требуется найти их наименьшего общего предка.


Определение:
Наименьшим общим предком (англ. least common ancestor) двух узлов [math]u[/math] и [math]v[/math] в корневом дереве [math]T[/math] называется узел [math]w[/math], который среди всех узлов, являющихся предками как узла [math]u[/math], так и [math]v[/math], имеет наибольшую глубину.


Алгоритм

Идея

Будем решать задачу [math]LCA[/math], уже умея решать задачу [math]RMQ[/math]. Тогда поиск наименьшего общего предка [math]i[/math]-того и [math]j[/math]-того элементов сводится к запросу минимума на отрезке массива, который будет введен позднее.

Препроцессинг

Для каждой вершины [math]T[/math] определим глубину с помощью следующей рекурсивной формулы:

[math]\mathrm{depth}(u) = \begin{cases} 0 & u = \mathrm{root}(T),\\ \mathrm{depth}(v) + 1 & u = \mathrm{son}(v). \end{cases}[/math]

Ясно, что глубина вершины элементарным образом поддерживается во время обхода в глубину.

Запустим обход в глубину из корня, который будет вычислять значения следующих величин:

  1. Cписок глубин посещенных вершин [math]d[/math]. Глубина текущей вершины добавляется в конец списка при входе в данную вершину, а также после каждого возвращения из её сына.
  2. Список посещений узлов [math]\mathtt{vtx}[/math], строящийся аналогично предыдущему, только добавляется не глубина а сама вершина.
  3. Значение функции [math]\mathtt{I}[u][/math], возвращающей индекс в списке глубин [math]d[/math], по которому была записана глубина вершины [math]u[/math] (например на момент входа в вершину).

Вот таким образом будут выглядеть эти три массива после обхода в глубину:

Пример массива [math]\mathtt{vtx}[/math]


Запрос

Будем считать, что [math]\mathrm{rmq}(d,l,r)[/math] возвращает индекс минимального элемента в [math]d[/math] на отрезке [math][l..r][/math]. Тогда ответом на запрос [math]\mathrm{lca}(u, v)[/math], где [math]\mathtt{I}[u] \leqslant \mathtt{I}[v][/math], будет [math]\mathtt{vtx}[\mathrm{rmq}(d,\mathtt{I}[u], \mathtt{I}[v])][/math].

Доказательство корректности алгоритма

Теорема:
Наименьшему общему предку вершин [math]u, v[/math] соответствует минимальная глубина на отрезке [math]d[\mathtt{I}[u], \mathtt{I}[v]][/math].
Доказательство:
[math]\triangleright[/math]
Рассмотрим два узла [math]u, v[/math] корневого дерева [math]T[/math]. Рассмотрим отрезок [math]d[\mathtt{I}[u]..\mathtt{I}[v]][/math]. Поскольку этот отрезок — путь из [math]u[/math] в [math]v[/math], он проходит через их наименьшего общего предка [math]w[/math] (в дереве есть только один простой путь между вершинами), а следовательно минимум на отрезке никак не больше глубины [math]w[/math]. Заметим, что в момент добавления [math]\mathtt{I}[u][/math] обход посещал поддерево с корнем [math]w[/math]. В момент добавления [math]\mathtt{I}[v][/math] мы все еще в поддереве с корнем [math]w[/math]. Значит, и на отрезке между [math]\mathtt{I}[u][/math] и [math]\mathtt{I}[v][/math] мы находились внутри поддерева с корнем [math]w[/math]. Отсюда сделаем заключение, что на рассматриваемом отрезке не посещалась вершина, отличная от [math]w[/math], с глубиной меньшей либо равной глубины [math]w[/math], т. к. подобной вершины нет в поддереве с корнем [math]w[/math].
[math]\triangleleft[/math]
.

Пример

Рассмотрим дерево на рисунке 1. Найдем наименьшего общего предка вершин, помеченных красным цветом. Список глубин, получающийся в результате обхода в глубину — [math][0, 1, 2, 1, 2, 1, 0, 1, 0].[/math] Глубина наименьшего общего предка красных вершин равна минимуму на отрезке [math][2, 1, 0, 1].[/math]

Рисунок к примеру

Сложность

Для нахождения минимального элемента на отрезке можно использовать алгоритм Фарака-Колтона и Бендера для [math]\pm 1RMQ[/math], т. к. соседние элементы в списке глубин отличаются не более чем на единицу. Длина списка глубин составляет [math](2n - 1)[/math], таким образом, препроцессинг работает за [math]O(n)[/math]. Время выполнения запроса равно времени запроса минимального элемента на отрезке — [math]O(1)[/math].

См.также

Источники информации