Статистики на отрезках. Корневая эвристика

Корневая эвристика (Sqrt-декомпозиция) — это структура данных, которая позволяет выполнять ассоциативные операции (например, суммирование элементов, нахождение минимума/максимума и т.д.) над элементами некоторого множества за [math] O(\sqrt n)[/math].

Построение

Пусть дан массив [math]A[/math] размерности [math]n[/math]. Cделаем следующие действия:

• разделим массив [math]A[/math] на блоки длины [math]len = \lfloor \sqrt{n} \rfloor[/math] ,
• в каждом блоке заранее посчитаем необходимую операцию,
• результаты подсчета запишем в массив [math]B[/math] размерности [math]cnt[/math], где [math]cnt = \left\lceil \dfrac{n}{len} \right\rceil[/math] — количество блоков.

Пример реализации построения массива [math]B[/math] для операции [math] \circ [/math]:

build()
    for i = 0 ... cnt
        B[i] = neutral                // neutral — нейтральный элемент для операции [math] \circ [/math] 
    for i = 0 ... n - 1
        B[i / len] = B[i / len] [math] \circ [/math] A[i]

Построение, очевидно, происходит за [math]O(n)[/math] времени.

Обработка запроса

Пусть получен запрос на выполнение операции на отрезке [math][l, r][/math]. Отрезок может охватить некоторые блоки массива [math]B[/math] полностью, а так же не более двух блоков (начальный и конечный) — не полностью.

Таким образом, для того чтобы найти результат операции на отрезке [math][l, r][/math] необходимо вручную выполнить ее на "хвостах", а потом выполнить ее для полученного результата и полных блоков, значения которых мы посчитали заранее.

Пример реализации обработки запроса:

[math] \circ [/math] — операция, для которой было сделано построение.

query(int l, int r)
    left = l / len
    right = r / len
    end = (left + 1) * len - 1
    res = neutral                       //neutral — нейтральный элемент для операции [math] \circ [/math] 
    if left == right
        for i = l ... r
            res = res [math] \circ [/math] A[i]
    else
        for i = l ... end
            res = res [math] \circ [/math] A[i]
        for i = left + 1 ... right - 1
            res = res [math] \circ [/math] B[i]
        for i = right * len ... r
            res = res [math] \circ [/math] A[i]

Размер каждого из "хвостов", очевидно, не превосходит длины блока [math]len[/math], а количество блоков не превосходит [math]cnt[/math]. Поскольку [math]len[/math] было выбрано равным [math]\lfloor \sqrt{n} \rfloor[/math] , а [math]cnt[/math] было выбрано равным [math]\left\lceil \dfrac{n}{len} \right\rceil[/math] , то для выполнения операции на отрезке [math][l, r][/math] понадобится [math]O(\sqrt{n})[/math] времени.

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

Реализация данного запроса будет зависеть от того, имеет ли операция, для которой сделано построение, обратную операцию и обладает ли она свойством коммутативности.

• если оба условия выполняются, то запрос на изменение элемента можно сделать за [math]O(1)[/math] времени;
• если хотя бы одно из условий не выполняется, то запрос на изменение элемента можно сделать за [math]O(\sqrt{n})[/math] времени.

Примеры реализации:

[math]p[/math] — номер элемента из массива [math]A[/math], который необходимо заменить; [math]newValue[/math] — новое значение для данного элемента.

Запрос на изменение элемента для операции, у которой есть обратная операция, и выполняется свойство коммутативности:

set(int p, value newValue)
    tmp = B[p / len] [math] \circ [/math] inverse(A[p])   // inverse(A[p]) — обратный элемент
    A[p] = newValue
    B[p / len] = tmp [math] \circ [/math] newValue

Запрос на изменение элемента для операции, у которой хотя бы одно из условий не выполняется:

set(int p, value newValue)
    index = len * (p / len)
    A[p] = newValue
    B[p / len] = neutral              // neutral — нейтральный элемент для операции [math] \circ [/math] 
    for i = index ... index + len - 1
        B[p / len] = B[p / len] [math] \circ [/math] A[i]

См. также

• Дерево отрезков. Построение
• Многомерное дерево отрезков

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

• Maximal:: algo:: Sqrt - декомпозиция
• Sqrt-декомпозиция (корневая оптимизация)