Дерево Фенвика — различия между версиями

Материал из Викиконспекты
Перейти к: навигация, поиск
Строка 1: Строка 1:
 
{{Определение
 
{{Определение
 
|definition=
 
|definition=
'''Дерево Фе&#769;нвика (Binary indexed tree)''' — структура данных, требующая <tex> O(n) </tex> памяти и позволяющая эффективно (за <tex> O(\log n) </tex>)
+
'''Дерево Фе&#769;нвика''' (англ. ''Binary indexed tree'') — структура данных, требующая <tex> O(n) </tex> памяти и позволяющая эффективно (за <tex> O(\log n) </tex>)
 
# изменять значение любого элемента в массиве;
 
# изменять значение любого элемента в массиве;
 
# выполнять некоторую ассоциативную, коммутативную, обратимую операцию <tex> G </tex> на отрезке <tex> [i, j] </tex>.
 
# выполнять некоторую ассоциативную, коммутативную, обратимую операцию <tex> G </tex> на отрезке <tex> [i, j] </tex>.
Строка 9: Строка 9:
  
 
Пусть дан массив <tex> A </tex> из <tex> n </tex> элементов: <tex> a_i, i = 0 .. n - 1 </tex>.<br/>
 
Пусть дан массив <tex> A </tex> из <tex> n </tex> элементов: <tex> a_i, i = 0 .. n - 1 </tex>.<br/>
Деревом Фенвика будем называть массив <tex> T </tex> из <tex> n </tex> элементов: <tex> T_i = \sum\limits_{k = F(i)}^{i} a_k, i = 0 .. n - 1 </tex>, где <tex> F(i) </tex> - некоторая функция.
+
Деревом Фенвика будем называть массив <tex> T </tex> из <tex> n </tex> элементов: <tex> T_i = \sum\limits_{k = F(i)}^{i} a_k, i = 0 .. n - 1 </tex>, где <tex> F(i) </tex> некоторая функция.
 
От выбора функции зависит время работы операций над деревом. Рассмотрим функцию, позволяющую делать обе операции за время <tex> O(\log n) </tex>.
 
От выбора функции зависит время работы операций над деревом. Рассмотрим функцию, позволяющую делать обе операции за время <tex> O(\log n) </tex>.
  
 
<tex> F(i) = i - 2^{h(i)} + 1, </tex> где <tex> h(i) </tex> - количество единиц в конце бинарной записи числа <tex> i </tex>. Эту функцию можно записать так: <tex> F(i) = i - t(i), </tex> где <tex> t(i) = 2^{h(i)} - 1, </tex> то есть число, состоящее из <tex> i </tex> единиц. Это значит, что функция заменяет все подряд идущие единицы в конце числа на нули.
 
<tex> F(i) = i - 2^{h(i)} + 1, </tex> где <tex> h(i) </tex> - количество единиц в конце бинарной записи числа <tex> i </tex>. Эту функцию можно записать так: <tex> F(i) = i - t(i), </tex> где <tex> t(i) = 2^{h(i)} - 1, </tex> то есть число, состоящее из <tex> i </tex> единиц. Это значит, что функция заменяет все подряд идущие единицы в конце числа на нули.
  
Эта функция задается простой формулой: <tex> F(i) = i \And (i + 1) </tex>, где <tex> \And </tex> — это операция побитового логического "И". При побитовом "И" числа и этого числа, увеличенного на единицу, мы получаем число без последних подряд идущих единиц.
+
Эта функция задается простой формулой: <tex> F(i) = i \And (i + 1) </tex>, где <tex> \And </tex> — это операция побитового логического "И". При побитовом "И" числа и его значения, увеличенного на единицу, мы получаем это число без последних подряд идущих единиц.
  
 
== Запрос изменения элемента ==
 
== Запрос изменения элемента ==
Строка 46: Строка 46:
 
Напишем функцию, которая будет изменять элемент <tex>a_i</tex> на <tex>d</tex>, и при этом меняет соответствующие частичные суммы.
 
Напишем функцию, которая будет изменять элемент <tex>a_i</tex> на <tex>d</tex>, и при этом меняет соответствующие частичные суммы.
  
  modify(i, d):
+
  '''function''' modify(i, d):
     while i < N
+
     '''while''' i < N
 
         t[i] += d
 
         t[i] += d
 
         i = i | (i + 1)
 
         i = i | (i + 1)
Строка 72: Строка 72:
 
=== Реализация ===
 
=== Реализация ===
  
Приведем код функции <tex> \mathrm sum(i) </tex> на C++:
+
Приведем код функции <tex> \mathrm sum(i) </tex>:
  <code>
+
  '''int''' sum(i):
int sum(int i)
+
     result = 0;
{
+
     '''while''' i >= 0
     int result = 0;
 
     while (i >= 0)
 
    {
 
 
         result += t[i];
 
         result += t[i];
 
         i = f(i) - 1;
 
         i = f(i) - 1;
     }
+
     '''return''' result;
    return result;
 
}
 
</code>
 
  
 
==Преимущества и недостатки дерева Фенвика==
 
==Преимущества и недостатки дерева Фенвика==

Версия 20:09, 16 апреля 2015

Определение:
Дерево Фе́нвика (англ. Binary indexed tree) — структура данных, требующая [math] O(n) [/math] памяти и позволяющая эффективно (за [math] O(\log n) [/math])
  1. изменять значение любого элемента в массиве;
  2. выполнять некоторую ассоциативную, коммутативную, обратимую операцию [math] G [/math] на отрезке [math] [i, j] [/math].
По горизонтали - содержимое массива T,
по вертикали - содержимое массива A

Впервые описано Питером Фенвиком в 1994 году.

Пусть дан массив [math] A [/math] из [math] n [/math] элементов: [math] a_i, i = 0 .. n - 1 [/math].
Деревом Фенвика будем называть массив [math] T [/math] из [math] n [/math] элементов: [math] T_i = \sum\limits_{k = F(i)}^{i} a_k, i = 0 .. n - 1 [/math], где [math] F(i) [/math] — некоторая функция. От выбора функции зависит время работы операций над деревом. Рассмотрим функцию, позволяющую делать обе операции за время [math] O(\log n) [/math].

[math] F(i) = i - 2^{h(i)} + 1, [/math] где [math] h(i) [/math] - количество единиц в конце бинарной записи числа [math] i [/math]. Эту функцию можно записать так: [math] F(i) = i - t(i), [/math] где [math] t(i) = 2^{h(i)} - 1, [/math] то есть число, состоящее из [math] i [/math] единиц. Это значит, что функция заменяет все подряд идущие единицы в конце числа на нули.

Эта функция задается простой формулой: [math] F(i) = i \And (i + 1) [/math], где [math] \And [/math] — это операция побитового логического "И". При побитовом "И" числа и его значения, увеличенного на единицу, мы получаем это число без последних подряд идущих единиц.

Запрос изменения элемента

Лемма:
Нам надо научиться быстро изменять частичные суммы в зависимости от того, как изменяются элементы. Рассмотрим как изменять величину [math]a_{k}[/math] на величину [math]d[/math]. Необходимо изменить элементы дерева [math]T_{i}[/math], для которых верно неравенство [math]F(i) \leqslant k \leqslant i[/math] .
Доказательство:
[math]\triangleright[/math]
[math] T_i =\sum\limits_{k = F(i)}^{i} a_k , i = 0 .. n - 1 \Rightarrow[/math] необходимо менять те [math]i[/math], для которых [math]a_{k}[/math] попадает в [math]T_i \Rightarrow[/math] необходимые [math] i [/math] удовлетворяют условию [math]F(i) \leqslant k \leqslant i[/math].
[math]\triangleleft[/math]
Лемма:
Можно перебрать все [math] i [/math], попадающие под неравенство по формуле [math]i_{next} = i_{prev} \mid (i_{prev} + 1) [/math], где [math] \mid [/math] — это операция побитового логического "ИЛИ".
Доказательство:
[math]\triangleright[/math]
Первый элемент последовательности само [math] k [/math]. Для него выполняется равенство, так как [math] F(i) \lt i [/math]. По формуле [math]i_{next} = i_{prev} \mid (i_{prev} + 1) [/math] мы заменим первый ноль на единицу. Неравенство при этом сохранится, так как [math]F(i)[/math] осталось прежним или уменьшилось, а [math] i [/math] увеличилось. Можем заметить, что если количество единиц в конце не будет совпадать с [math] k [/math], то формула [math]i_{next} = i_{prev} \mid (i_{prev} + 1) [/math] нарушит неравенство, потому что либо само [math] i [/math] будет меньше, чем k, либо [math] F(i) [/math] станет больше, чем [math] k [/math]. Таким образом, перебраны будут только нужные элементы
[math]\triangleleft[/math]

Все [math]i[/math] мы можем получить следующим образом : [math]i_{next} = i_{prev} \mid (i_{prev} + 1) [/math]. Следующим элементом в последовательности будет элемент, у которого первый с конца ноль превратится в единицу. Можно заметить, что если к исходному элементу прибавить единицу, то необходимый ноль обратится в единицу, но при этом все следующие единицы обнулятся. Чтобы обратно их превратить в единицы, применим операцию побитового ИЛИ. Таким образом все нули в конце превратятся в единицы и мы получим нужный элемент. Для того, чтобы понять, что эта последовательность верна, достаточно посмотреть на таблицу.

[math]\i_{prev}[/math] [math]\cdots 011 \cdots 1[/math]
[math]i_{prev} + 1[/math] [math]\cdots 100 \cdots 0[/math]
[math]i_{next}[/math] [math]\cdots 111 \cdots 1[/math]


Несложно заметить, что данная последовательность строго возрастает и в худшем случае будет применена логарифм раз, так как добавляет каждый раз по одной единице в двоичном разложении числа [math]i[/math]. Напишем функцию, которая будет изменять элемент [math]a_i[/math] на [math]d[/math], и при этом меняет соответствующие частичные суммы.

function modify(i, d):
   while i < N
       t[i] += d
       i = i | (i + 1)

Запрос получения суммы на префиксе

В качестве бинарной операции [math] G [/math] рассмотрим операцию сложения.
Обозначим [math] G_i = \mathrm sum(i) = \sum\limits_{k = 0}^{i} a_k [/math]. Тогда [math] \mathrm sum(i, j) = \sum\limits_{k = i}^{j} a_k = G_j - G_{i - 1} [/math].

Лемма:
[math] a_i [/math] входит в сумму для [math] t_k [/math], если [math] \exists j: k = i \mid (2^j - 1) [/math].

Для доказательства леммы рассмотрим битовую запись следующих чисел: [math] k - 2^{h(k)} + 1 \leqslant i \leqslant k [/math]

[math]k - 2^{h(k)} + 1[/math] [math]\cdots (0 \cdots 0)[/math]
[math]i[/math] [math]\cdots (\cdots \cdots)[/math]
[math]k[/math] [math]\cdots (1 \cdots 1)[/math]

Реализация

Приведем код функции [math] \mathrm sum(i) [/math]:

int sum(i):
   result = 0;
   while i >= 0
       result += t[i];
       i = f(i) - 1;
   return result;

Преимущества и недостатки дерева Фенвика

Главными преимуществами данной конструкции являются простота реализации и быстрота ответов на запросы за [math] O(1) [/math]. Также дерево Фенвика позволяет быстро изменять значения в массиве и находить некоторые функции от элементов массива. Недостатком является то, что при изменении одного элемента исходного массива, приходится пересчитывать частичные суммы, а это затратно по времени.

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