Многомерное дерево отрезков — различия между версиями

Материал из Викиконспекты
Перейти к: навигация, поиск
м
м (rollbackEdits.php mass rollback)
 
(не показаны 53 промежуточные версии 8 участников)
Строка 1: Строка 1:
[[Дерево отрезков. Построение|Дерево отрезков]] естественным образом обобщается на двумерный и вообще говоря многомерный случай. Такая структура данных может вычислять значение некоторой [[Ассоциативная_операция|ассоциативной функции]] на гиперпрямоугольнике за <tex>O(\log^{p} n)</tex>, где <tex>p</tex> {{---}} размерность пространства, а <tex>n</tex> {{---}} ширина гиперкуба на котором производятся вычисления.
+
[[Дерево отрезков. Построение|Дерево отрезков]] естественным образом обобщается на двумерный и, вообще говоря, многомерный случай. Такая структура данных может вычислять значение некоторой [[Ассоциативная_операция|ассоциативной функции]] на гиперпрямоугольнике. Например, она позволяет решать следующую задачу.{{Задача|definition = Дан <tex>p</tex>-мерный массив, где индекс каждого измерения массива может принимать значения от <tex>1</tex> до <tex>n</tex>. Необходимо уметь изменять значение элемента массива, а также находить сумму на <tex>p</tex>-мерной области. }}  
 +
Каждую из этих операций многомерное дерево отрезков выполняет за <tex>O(\log^{p} n)</tex>.
  
==Структура==
+
К примерам задач, решаемых с помощью многомерного дерева отрезков, также можно отнести задачи, которые решаются с помощью одномерного [[Дерево отрезков. Построение|дерева отрезков]], только теперь в многомерном случае, а еще ,например, задачу поиска числа точек в заданном прямоугольнике, которую иначе можно решать при помощи [[Перечисление точек в произвольном прямоугольнике за n * log ^(d - 1) n (range tree)|range tree]], и другие.
<tex>n</tex>-мерное дерево отрезков {{---}} обычное дерево отрезков, элементами которого являются деревья отрезков размерности на 1 меньше. Основная идея заключается в рекурсивном переходе к деревьям меньшей размерности. Рассмотрим работу этого принципа на следующем примере. Пусть задано <tex>p</tex>-мерное пространство с координатными осями <tex>x_1, x_2, x_3...x_p</tex>. Необходимо вычислять некоторую ассоциативную функцию на гиперпрямоугольнике. Для этого сначала найдем элементы дерева, соответствующие <tex>x_1</tex> координате. Для каждого из этих элементов рекурсивно перейдем в соответствующие им деревья отрезков и в них найдем элементы, отвечающие соответствующим координатам <tex>x_2</tex> исходной гиперпрямоугольной области, и т. д. Таким образом, алгоритм совершит <tex>p</tex> вхождений в рекурсию, каждая итерация которой работает за <tex>O(\log n)</tex> и получим необходимую асимптотику.
 
  
==Двумерное дерево отрезков==
+
==Принцип работы==
Поскольку в большинстве случаев на практике используются деревья отрезков размерности не более 2, а также для облегчения понимания далее будут рассматриваться только двумерные деревья отрезков. Рассмотрим следующую задачу.
+
[[Файл:SegmentTreeWorking.png|thumb|600px|right|Пример некоторой стадии работы алгоритма (поиск элементов, подходящих некоторой области)]]
{{Задача
+
<tex>n</tex>-мерное дерево отрезков {{---}} обычное дерево отрезков, элементами которого являются деревья отрезков размерности на единицу меньше. Основная идея заключается в рекурсивном переходе к деревьям меньшей размерности. Рассмотрим работу этого принципа на следующем примере. Пусть задано <tex>p</tex>-мерное пространство с координатными осями <tex>x_1, x_2, x_3 \ldots x_p</tex>. Необходимо найти значение некоторой ассоциативной функции на гиперпрямоугольнике.
|definition=
+
 
Дано поле размером <tex>N \times M</tex>, заполненное некоторыми числами. Необходимо уметь обрабатывать два типа запросов:
+
Функция, вычисляющая ответ, должна работать следующим образом. На вход она принимает <tex>i</tex>-мерное дерево отрезков, которое соответствует рассматриваемой области (где <tex>i</tex> {{---}} количество координатных осей, которые не были рассмотрены), а также <tex>i</tex>-мерную область, для которой следует вычислить функцию. Вначале она находит <tex>i-1</tex>-мерные деревья отрезков, которые соответствуют отрезку по <tex>p-i+1</tex> координате, и рекурсивно запускается от них (если текущее дерево одномерное, то функция просто возвращает ответ из соответствующего листа). После этого считает итоговый результат, используя полученные после рекурсивных вызовов значения.
* Изменить некоторый элемент
+
 
* Посчитать сумму на прямоугольной области
+
Для того, чтобы определить, от каких именно деревьев отрезков следует запускаться рекурсивно, действовать необходимо так же, как и в одномерном случае. Т. е. если текущий отрезок не пересекается с необходимым, то возвращаем нейтральный элемент, если он полностью лежит в необходимом отрезке, то рекурсивно переходим к следующей координате, иначе разобьем текущий отрезок пополам, и рассмотри отдельно каждую из частей.  
}}
+
 
 +
На рисунке справа показан пример обработки очередной координаты (поиск соответствующих отрезку элементов {{---}} деревьев на <tex>1</tex> меньшей мерности).
 +
 
 +
Таким образом, алгоритм совершит <tex>p</tex> вхождений в рекурсию, каждая итерация которой работает за <tex>O(\log n)</tex> и получим необходимую асимптотику.
  
 
==Хранение==
 
==Хранение==
[[Файл:SegmentTree2DExample.png|thumb|250px|right|Пример двумерного дерева отрезков для 16 элементов]]
+
[[Файл:SegmentTree2DExample.png|thumb|350px|right|Пример двумерного дерева отрезков для 16 элементов]]
Двумерное дерево отрезков удобно хранить в виде массива, размером <tex>4N \times 4M</tex>. Каждая строчка такого массива соответствует некоторому отрезку по первой координате. Сама же строчка является деревом отрезков по второй координате.
+
Пусть необходимо хранить дерево отрезков для <tex>p</tex>-мерной области, размеры которой <tex>n_1 \times n_2 \times \ldots \times n_p</tex>. Удобнее всего это делать с помощью <tex>p</tex>-мерного массива. Однако его размеры по каждой координате, так же как и в одномерном случае, должны превышать размеры соответствующего отрезка в 4 раза. На самом деле нам нужно хранить <tex>2n</tex> чисел, но, если мы хотим, чтобы правый и левый сын некоторой вершины <tex>i</tex> находились на <tex> 2i + 1</tex> и <tex>2i + 2</tex> месте, то, если длина отрезка не является степенью двойки, некоторые элементы массива могут быть не задействованы, поэтому в худшем случае, может понадобиться массив, размер которого в 4 раза превышает количество элементов. Т. е. потребуется массив размером <tex>4 n_1 \times 4 n_2 \times \ldots \times 4 n_p</tex>. Так двумерное дерево отрезков удобно хранить в виде массива, размером <tex>4N \times 4M</tex>. Каждая строчка такого массива соответствует некоторому отрезку по первой координате. Сама же строчка является деревом отрезков по второй координате.
  
На рисунке справа показан пример дерева отрезков для массива 4 на 4, заполненного числами от 1 от 16. Например, в элементе <tex>a[2][0] = 100</tex> хранится сумма элементов, соответствующих отрезку <tex>[2..3]</tex> по первой координате и <tex>[0..3]</tex> по второй в исходном массиве. А в ячейке <tex>a[0][0] = 136</tex> хранится сумма всех элементов.
+
На рисунке справа показан пример дерева отрезков для суммы на массиве 4 на 4, заполненного числами от 1 от 16. Например, в элементе <tex>a[2][0] = 100</tex> хранится сумма элементов, соответствующих отрезку <tex>[2..3]</tex> по первой координате и <tex>[0..3]</tex> по второй в исходном массиве. А в ячейке <tex>a[0][0] = 136</tex> хранится сумма всех элементов.
  
 
Интересно, что если построить дерево вначале по второй координате, а потом по первой, то получившийся массив будет таким же. Т. е. данный двумерный массив можно рассматривать как массив деревьев отрезков, где каждое дерево соответствует некоторому отрезку по второй координате, а в нем хранятся суммы по первой.
 
Интересно, что если построить дерево вначале по второй координате, а потом по первой, то получившийся массив будет таким же. Т. е. данный двумерный массив можно рассматривать как массив деревьев отрезков, где каждое дерево соответствует некоторому отрезку по второй координате, а в нем хранятся суммы по первой.
Строка 23: Строка 26:
 
Заметим, что в общем случае для хранения <tex>p</tex>-мерного дерева отрезков требуется <tex>4^p n</tex> памяти, где <tex>n</tex> {{---}} общее количество элементов.
 
Заметим, что в общем случае для хранения <tex>p</tex>-мерного дерева отрезков требуется <tex>4^p n</tex> памяти, где <tex>n</tex> {{---}} общее количество элементов.
  
==Псевдокод==
+
==Запрос==
Построение:
+
Рассмотрим отличия реализации многомерного и одномерного случаев. На самом деле, отличаются реализации только в двух местах. Во-первых, если рассматриваемый отрезок совпадает с необходимым, то в одномерном случае функция просто возвращает число, которое находится в текущем элементе массива. В многомерном случае, если рассматриваемая координата не последняя, следует вместо этого узнать значение, рекурсивно перейдя к следующей координате, и вернуть его.
// first call - buildX(0, 0, n, a, t)
 
// a - исходный массив
 
// t - массив дерева отрезков
 
// [lx, rx) - полуинтервал
 
 
buildX(vx, lx, rx, a[][], t[][])
 
  if lx != rx
 
    mx = (lx + rx + 1) / 2
 
    buildX(vx * 2 + 1, lx, mx, a, t)
 
    buildX(vx * 2 + 2, mx, rx, a, t)
 
  buildY(vx, lx, rx, 0, 0, m, a, t)
 
 
buildY(vx, lx, rx, vy, ly, ry, a[][], t[][])
 
  if ly == ry
 
    if lx == rx
 
      t[vx][vy] = a[lx][ly]
 
    else
 
      t[vx][vy] = t[vx * 2 + 1][vy] + t[vx * 2 + 2][vy]
 
  else
 
    my = (ly + ry + 1) / 2
 
    buildY(vx, lx, rx, vy * 2 + 1, ly, my, a, t)
 
    buildY(vx, lx, rx, vy * 2 + 2, my, ry, a, t)
 
    t[vx][vy] = t[vx][vy * 2 + 1] + t[vx][vy * 2 + 2]
 
  
Подсчет суммы элементов:
+
Еще один момент, в которых отличается реализация {{---}} передаваемые в функцию параметры. В многомерном случае кроме всего прочего следует также передать рассматриваемое <tex>p-i+1</tex>-мерное дерево (или кортеж из чисел, указывающих на соответствующие элементы массива), а также область, которую следует рассматривать (или <tex>p-i+1</tex> пар чисел, обозначающих отрезки на соответствующих координатных осях). Все остальные детали реализации остаются такими же как и в одномерном дереве отрезков.
// first call - sumX(0, 0, n - 1, lx, rx, ly, ry, t)
 
 
sumX(vx, tlx, trx, lx, rx, ly, ry, t[][])
 
  if lx > rx
 
  return 0
 
  if lx == tlx && rx == trx
 
  return sumY(vx, 0, 0, m - 1, ly, ry)
 
  tmx = (tlx + trx) / 2
 
  return sumX(vx * 2 + 1, tlx, tmx, lx, min(rx, tmx), ly, ry, t) +
 
        sumX(vx * 2 + 2, tmx + 1, trx, max(lx, tmx + 1), rx, ly, ry, t)
 
 
sumY(vx, vy, tly, try, ly, ry)
 
  if ly > ry
 
  return 0
 
  if ly == tly && ry == try
 
  return t[vx][vy]
 
  tmy = (tly + try) / 2
 
  return sumY(vx, vy * 2 + 1, tly, tmy, ly, min(ry, tmy), t) +
 
        sumY(vx, vy * 2 + 2, tmy + 1, try, max(ly, tmy + 1), ry, t)
 
  
Обновление элемента:
+
В каждом нижеприведенном псевдокоде будут встречены обозначения:
// first call - updateX(0, 0, n - 1, x, y, val, t)
+
* индекс <tex>\mathtt{P}</tex> {{---}} размерность массива из условия задачи,
+
* <tex>\mathtt{\odot}</tex> {{---}} та операция, которую мы считаем на данном многомерном дереве отрезков.
updateX(vx, lx, rx, x, y, val, t[][])
 
  if lx != rx
 
  mx = (lx + rx) / 2
 
  if x <= mx
 
    updateX(vx * 2 + 1, lx, mx, x, y, val, t)
 
  else
 
    updateX(vx * 2 + 2, mx + 1, rx, x, y, val, t)
 
  updateY(vx, lx, rx, 0, 0, m - 1, x, y, val, t)
 
 
updateY(vx, lx, rx, vy, ly, ry, x, y, val, t[][])
 
  if ly == ry
 
  if lx == rx
 
    t[vx][vy] = val
 
  else
 
    t[vx][vy] = t[vx * 2 + 1][vy] + t[vx * 2 + 2][vy]
 
  else
 
  my = (ly + ry) / 2
 
  if y <= my
 
    updateY(vx, lx, rx, vy * 2 + 1, ly, my, x, y, val, t)
 
  else
 
    updateY(vx, lx, rx, vy * 2 + 2, my + 1, ry, x, y, val, t)
 
  t[vx][vy] = t[vx][vy * 2 + 1] + t[vx][vy * 2 + 2] 
 
  
 +
В нижеприведенном псевдокоде будет встречен <tex>\varepsilon</tex>  {{---}} нейтральный элемент.
  
==Источники==
+
 
* [http://e-maxx.ru/algo/segment_tree Дерево отрезков {{---}} e-maxx.ru]
+
Псевдокод:
* [http://habrahabr.ru/post/131072/)/ Двумерное дерево отрезков {{---}} habrahabr.ru]
+
<code>
 +
'''void''' query('''int''' area[], '''int''' x1, '''int''' x2, ..., '''int''' xP, '''int''' leftBorder, '''int''' rightBorder, '''int''' queryLeft, '''int''' queryRight, '''int''' node)
 +
    '''if''' queryLeft > queryRight
 +
        '''return''' <tex>\varepsilon</tex>
 +
    '''if''' leftBorder == queryLeft '''and''' rightBorder == queryRight
 +
        '''if''' последняя координата
 +
            '''return''' t[x1][x2]...[xP][node]
 +
        '''else'''
 +
            '''return''' query(area[], x1, x2, ..., xP, node, 0, m - 1, area[P + 2].left, area[P + 2].right, 0)
 +
    med = (leftBorder + rightBorder) / 2
 +
    '''return''' query(area[], x1, x2, ..., xP, leftBorder, med, queryLeft, min(queryRight, med), node * 2 + 1) <tex>\odot</tex>
 +
            query(area[], x1, x2, ..., xP, med + 1, rightBorder, max(queryLeft, med + 1), queryRight, node * 2 + 2)
 +
</code>
 +
 
 +
==Обновление==
 +
Как и в одномерном случае, обновить в массиве необходимо не один элемент, а все, которые отвечают за области, в которых он присутствует. Таким образом, при обработке отрезка по некоторой координате (если она не последняя) следует выполнить следующие действия:
 +
* Если рассматриваемый отрезок содержит больше одного элемента, разобьем его на две части и рекурсивно перейдем в ту, где находится необходимый элемент
 +
* Перейдем к следующей координате
 +
Заметим, что "переходов к следующей координаты" при рассмотрении некоторой координатной оси будет совершено <tex>\log n</tex>, а итоговая сложность составит <tex>O(\log^{p} n)</tex>.
 +
 
 +
Отдельно следует рассмотреть, что происходит, когда текущее дерево является одномерным (мы рассмотрели все координаты, кроме текущей):
 +
* Если рассматриваемый отрезок содержит больше одного элемента, разобьем его на две части и рекурсивно перейдем в ту, где находится необходимый элемент
 +
* Найдем первую координату, в которой рассматривается больше одного элемента. Обновим значение элемента массива с помощью уже вычисленных значений для разбитого надвое отрезка по этой координате.
 +
* Если мы рассматриваем область, состоящую из одного элемента, обновим значение массива.
 +
 
 +
Псевдокод:
 +
<code>
 +
'''void''' update('''int''' newElem, '''int''' x1, '''int''' x2, ..., '''int''' xP, '''int''' x1Left, '''int''' x1Right, '''int''' x2Left, '''int''' x2Right, ..., '''int''' xPLeft, '''int''' xPRight,
 +
            '''int''' leftBorder, '''int''' rightBorder, '''int''' node)
 +
    '''if''' leftBorder != rightBorder
 +
        med = (leftBorder + rightBorder) / 2
 +
    '''if''' med >= newElem.x(P+1)
 +
        update(newElem, x1, x2, ..., xP, x1Left, x1Right, x2Left, x2Right, ..., xPLeft, xPRight, leftBorder, med, node * 2 + 1)
 +
    '''else'''
 +
        update(newElem, x1, x2, ..., xP, x1Left, x1Right, x2Left, x2Right, ..., xPLeft, xPRight, med + 1, rightBorder, node * 2 + 2)
 +
    '''if''' последняя координата
 +
        '''for''' I = 1..n
 +
            '''if''' xILeft != xIRigth
 +
                t[x1][x2]...[xP][node] = t[x1][x2]...[xI * 2 + 1]...[node] <tex>\odot</tex> t[x1][x2]...[xI * 2 + 2]...[node]
 +
                '''return'''
 +
        t[x1][x2]...[xP][node] = newElem.value
 +
    '''else'''
 +
        '''if''' leftBorder != rightBorder
 +
            update(newElem, x1, x2, ..., xP, node, x1Left, x1Rigth, x2Left, x2Right, ..., leftBorder, rightBorder, 0, m - 1, 0)
 +
</code>
 +
 
 +
==Построение==
 +
Построение многомерного дерева отрезков практически ничем не отличается от его обновления. Единственное различие {{---}} если рассматриваемый отрезок состоит из более чем одного элемента, то необходимо рекурсивно вызываться из обеих частей.
 +
 
 +
Псевдокод:
 +
<code>
 +
'''void''' build('''int''' x1, '''int''' x2, ..., '''int''' xP, '''int''' x1Left, '''int''' x1Right, '''int''' x2Left, '''int''' x2Right, ..., '''int''' xPLeft, '''int''' xPRight, '''int''' leftBorder,
 +
            '''int''' rightBorder, '''int''' node)
 +
    '''if''' leftBorder != rightBorder
 +
        med = (leftBorder + rightBorder) / 2
 +
        update(newElem, x1, x2, ..., xP, x1Left, x1Right, x2Left, x2Right, ..., xPLeft, xPRight, leftBorder, med, node * 2 + 1)
 +
        update(newElem, x1, x2, ..., xP, x1Left, x1Right, x2Left, x2Right, ..., xPLeft, xPRight, med + 1, rightBorder, node * 2 + 2)
 +
    '''if''' последняя координата
 +
        '''for''' I = 1..n
 +
            '''if''' xILeft != xIRight
 +
                t[x1][x2]...[xP][node] = t[x1][x2]...[xI * 2 + 1]...[node] <tex>\odot</tex> t[x1][x2]...[xI * 2 + 2]...[node]
 +
                '''return'''
 +
        t[x1][x2]...[xP][node] = data[x1Left][x2Left]...[xPLeft][node]
 +
    '''else'''
 +
        '''if''' leftBorder != rightBorder
 +
            update(newElem, x1, x2, ..., xP, node, x1Left, x1Rigth, x2Left, x2Right, ..., leftBorder, rightBorder, 0, m - 1, 0)
 +
</code>
 +
 
 +
Заметим, что построение дерева требует <tex>O(n)</tex> времени, где <tex>n</tex> {{---}} общее число элементов в массиве.
  
 
==См. также==
 
==См. также==
Строка 105: Строка 113:
 
*[[Сжатое многомерное дерево отрезков]]
 
*[[Сжатое многомерное дерево отрезков]]
 
*[[Многомерное дерево Фенвика]]
 
*[[Многомерное дерево Фенвика]]
 +
 +
==Источники информации==
 +
* [http://e-maxx.ru/algo/segment_tree MAXimal :: algo :: Дерево отрезков]
 +
* [http://habrahabr.ru/post/131072/)/ Habrahabr {{---}} Двумерное дерево отрезков]
 +
  
 
[[Категория: Дискретная математика и алгоритмы]]
 
[[Категория: Дискретная математика и алгоритмы]]
 
[[Категория: Дерево отрезков]]
 
[[Категория: Дерево отрезков]]
 +
[[Категория: Модификации структур данных]]

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

Дерево отрезков естественным образом обобщается на двумерный и, вообще говоря, многомерный случай. Такая структура данных может вычислять значение некоторой ассоциативной функции на гиперпрямоугольнике. Например, она позволяет решать следующую задачу.

Задача:
Дан [math]p[/math]-мерный массив, где индекс каждого измерения массива может принимать значения от [math]1[/math] до [math]n[/math]. Необходимо уметь изменять значение элемента массива, а также находить сумму на [math]p[/math]-мерной области.

Каждую из этих операций многомерное дерево отрезков выполняет за [math]O(\log^{p} n)[/math].

К примерам задач, решаемых с помощью многомерного дерева отрезков, также можно отнести задачи, которые решаются с помощью одномерного дерева отрезков, только теперь в многомерном случае, а еще ,например, задачу поиска числа точек в заданном прямоугольнике, которую иначе можно решать при помощи range tree, и другие.

Принцип работы

Пример некоторой стадии работы алгоритма (поиск элементов, подходящих некоторой области)

[math]n[/math]-мерное дерево отрезков — обычное дерево отрезков, элементами которого являются деревья отрезков размерности на единицу меньше. Основная идея заключается в рекурсивном переходе к деревьям меньшей размерности. Рассмотрим работу этого принципа на следующем примере. Пусть задано [math]p[/math]-мерное пространство с координатными осями [math]x_1, x_2, x_3 \ldots x_p[/math]. Необходимо найти значение некоторой ассоциативной функции на гиперпрямоугольнике.

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

Для того, чтобы определить, от каких именно деревьев отрезков следует запускаться рекурсивно, действовать необходимо так же, как и в одномерном случае. Т. е. если текущий отрезок не пересекается с необходимым, то возвращаем нейтральный элемент, если он полностью лежит в необходимом отрезке, то рекурсивно переходим к следующей координате, иначе разобьем текущий отрезок пополам, и рассмотри отдельно каждую из частей.

На рисунке справа показан пример обработки очередной координаты (поиск соответствующих отрезку элементов — деревьев на [math]1[/math] меньшей мерности).

Таким образом, алгоритм совершит [math]p[/math] вхождений в рекурсию, каждая итерация которой работает за [math]O(\log n)[/math] и получим необходимую асимптотику.

Хранение

Пример двумерного дерева отрезков для 16 элементов

Пусть необходимо хранить дерево отрезков для [math]p[/math]-мерной области, размеры которой [math]n_1 \times n_2 \times \ldots \times n_p[/math]. Удобнее всего это делать с помощью [math]p[/math]-мерного массива. Однако его размеры по каждой координате, так же как и в одномерном случае, должны превышать размеры соответствующего отрезка в 4 раза. На самом деле нам нужно хранить [math]2n[/math] чисел, но, если мы хотим, чтобы правый и левый сын некоторой вершины [math]i[/math] находились на [math] 2i + 1[/math] и [math]2i + 2[/math] месте, то, если длина отрезка не является степенью двойки, некоторые элементы массива могут быть не задействованы, поэтому в худшем случае, может понадобиться массив, размер которого в 4 раза превышает количество элементов. Т. е. потребуется массив размером [math]4 n_1 \times 4 n_2 \times \ldots \times 4 n_p[/math]. Так двумерное дерево отрезков удобно хранить в виде массива, размером [math]4N \times 4M[/math]. Каждая строчка такого массива соответствует некоторому отрезку по первой координате. Сама же строчка является деревом отрезков по второй координате.

На рисунке справа показан пример дерева отрезков для суммы на массиве 4 на 4, заполненного числами от 1 от 16. Например, в элементе [math]a[2][0] = 100[/math] хранится сумма элементов, соответствующих отрезку [math][2..3][/math] по первой координате и [math][0..3][/math] по второй в исходном массиве. А в ячейке [math]a[0][0] = 136[/math] хранится сумма всех элементов.

Интересно, что если построить дерево вначале по второй координате, а потом по первой, то получившийся массив будет таким же. Т. е. данный двумерный массив можно рассматривать как массив деревьев отрезков, где каждое дерево соответствует некоторому отрезку по второй координате, а в нем хранятся суммы по первой.

Заметим, что в общем случае для хранения [math]p[/math]-мерного дерева отрезков требуется [math]4^p n[/math] памяти, где [math]n[/math] — общее количество элементов.

Запрос

Рассмотрим отличия реализации многомерного и одномерного случаев. На самом деле, отличаются реализации только в двух местах. Во-первых, если рассматриваемый отрезок совпадает с необходимым, то в одномерном случае функция просто возвращает число, которое находится в текущем элементе массива. В многомерном случае, если рассматриваемая координата не последняя, следует вместо этого узнать значение, рекурсивно перейдя к следующей координате, и вернуть его.

Еще один момент, в которых отличается реализация — передаваемые в функцию параметры. В многомерном случае кроме всего прочего следует также передать рассматриваемое [math]p-i+1[/math]-мерное дерево (или кортеж из чисел, указывающих на соответствующие элементы массива), а также область, которую следует рассматривать (или [math]p-i+1[/math] пар чисел, обозначающих отрезки на соответствующих координатных осях). Все остальные детали реализации остаются такими же как и в одномерном дереве отрезков.

В каждом нижеприведенном псевдокоде будут встречены обозначения:

  • индекс [math]\mathtt{P}[/math] — размерность массива из условия задачи,
  • [math]\mathtt{\odot}[/math] — та операция, которую мы считаем на данном многомерном дереве отрезков.

В нижеприведенном псевдокоде будет встречен [math]\varepsilon[/math] — нейтральный элемент.


Псевдокод:

void query(int area[], int x1, int x2, ..., int xP, int leftBorder, int rightBorder, int queryLeft, int queryRight, int node)
    if queryLeft > queryRight
        return [math]\varepsilon[/math]
    if leftBorder == queryLeft and rightBorder == queryRight
        if последняя координата
            return t[x1][x2]...[xP][node]
        else
            return query(area[], x1, x2, ..., xP, node, 0, m - 1, area[P + 2].left, area[P + 2].right, 0)
    med = (leftBorder + rightBorder) / 2
    return query(area[], x1, x2, ..., xP, leftBorder, med, queryLeft, min(queryRight, med), node * 2 + 1) [math]\odot[/math]
           query(area[], x1, x2, ..., xP, med + 1, rightBorder, max(queryLeft, med + 1), queryRight, node * 2 + 2)

Обновление

Как и в одномерном случае, обновить в массиве необходимо не один элемент, а все, которые отвечают за области, в которых он присутствует. Таким образом, при обработке отрезка по некоторой координате (если она не последняя) следует выполнить следующие действия:

  • Если рассматриваемый отрезок содержит больше одного элемента, разобьем его на две части и рекурсивно перейдем в ту, где находится необходимый элемент
  • Перейдем к следующей координате

Заметим, что "переходов к следующей координаты" при рассмотрении некоторой координатной оси будет совершено [math]\log n[/math], а итоговая сложность составит [math]O(\log^{p} n)[/math].

Отдельно следует рассмотреть, что происходит, когда текущее дерево является одномерным (мы рассмотрели все координаты, кроме текущей):

  • Если рассматриваемый отрезок содержит больше одного элемента, разобьем его на две части и рекурсивно перейдем в ту, где находится необходимый элемент
  • Найдем первую координату, в которой рассматривается больше одного элемента. Обновим значение элемента массива с помощью уже вычисленных значений для разбитого надвое отрезка по этой координате.
  • Если мы рассматриваем область, состоящую из одного элемента, обновим значение массива.

Псевдокод:

void update(int newElem, int x1, int x2, ..., int xP, int x1Left, int x1Right, int x2Left, int x2Right, ..., int xPLeft, int xPRight, 
            int leftBorder, int rightBorder, int node)
    if leftBorder != rightBorder
        med = (leftBorder + rightBorder) / 2
    if med >= newElem.x(P+1)
        update(newElem, x1, x2, ..., xP, x1Left, x1Right, x2Left, x2Right, ..., xPLeft, xPRight, leftBorder, med, node * 2 + 1)
    else
        update(newElem, x1, x2, ..., xP, x1Left, x1Right, x2Left, x2Right, ..., xPLeft, xPRight, med + 1, rightBorder, node * 2 + 2)
    if последняя координата
        for I = 1..n
            if xILeft != xIRigth
                t[x1][x2]...[xP][node] = t[x1][x2]...[xI * 2 + 1]...[node] [math]\odot[/math] t[x1][x2]...[xI * 2 + 2]...[node]
                return
        t[x1][x2]...[xP][node] = newElem.value
    else
        if leftBorder != rightBorder
            update(newElem, x1, x2, ..., xP, node, x1Left, x1Rigth, x2Left, x2Right, ..., leftBorder, rightBorder, 0, m - 1, 0)

Построение

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

Псевдокод:

void build(int x1, int x2, ..., int xP, int x1Left, int x1Right, int x2Left, int x2Right, ..., int xPLeft, int xPRight, int leftBorder,
           int rightBorder, int node)
    if leftBorder != rightBorder
        med = (leftBorder + rightBorder) / 2
        update(newElem, x1, x2, ..., xP, x1Left, x1Right, x2Left, x2Right, ..., xPLeft, xPRight, leftBorder, med, node * 2 + 1)
        update(newElem, x1, x2, ..., xP, x1Left, x1Right, x2Left, x2Right, ..., xPLeft, xPRight, med + 1, rightBorder, node * 2 + 2)
    if последняя координата
        for I = 1..n
            if xILeft != xIRight
                t[x1][x2]...[xP][node] = t[x1][x2]...[xI * 2 + 1]...[node] [math]\odot[/math] t[x1][x2]...[xI * 2 + 2]...[node]
                return
        t[x1][x2]...[xP][node] = data[x1Left][x2Left]...[xPLeft][node]
    else
        if leftBorder != rightBorder
            update(newElem, x1, x2, ..., xP, node, x1Left, x1Rigth, x2Left, x2Right, ..., leftBorder, rightBorder, 0, m - 1, 0)

Заметим, что построение дерева требует [math]O(n)[/math] времени, где [math]n[/math] — общее число элементов в массиве.

См. также

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