Реализация встречного дерева Фенвика для RMQ или ДО для курильщиков
Difference between ru4 and ru5, changed 1,502 character(s)
Доброго времени суток!↵
Началось все с того, что после прочтения [статьи](https://neerc.ifmo.ru/wiki/index.php?title=%D0%92%D1%81%D1%82%D1%80%D0%B5%D1%87%D0%BD%D0%BE%D0%B5_%D0%B4%D0%B5%D1%80%D0%B5%D0%B2%D0%BE_%D0%A4%D0%B5%D0%BD%D0%B2%D0%B8%D0%BA%D0%B0) про встречное дерево Фенвика, я начал искать в интернете его реализацию, но (спойлер) так и не нашел. Безусловно, тема не нова, но, к сожалению, статью, описывающую эту структуру данных с приведенными примерами реализации, я так и не нашел. Перед прочтением данного поста советую ознакомиться с указанной статьей, т.к. на нее будут опираться все дальнейшие рассуждения.↵

Встречное дерево Фенвика↵
==================↵

Для начала, вспомним определение: Встречное дерево Фенвика (англ. counter tree Fenwick) — дерево Фенвика, в котором над каждым столбцом идет столбец такой же высоты, вычисляемый по формуле $f_{rev}[i]=\sum\limits_{j=i+1}^{i+2^
{h(i)}}a[j]$. Сразу сделаю уточнение, что под прямым деревом Фенвика подразумевается массив, посчитанный формуле: $f_{for}[i]=\sum\limits_{j=i-2^{h(i)}+1}^i a[j]$. Для пущей понятности, прикреплю картинку. ↵

![ ](https://neerc.ifmo.ru/wiki/images/0/0e/Vstbit.png)↵

Зная только то, что встречное дерево Фенвика симметрично прямому, мы можем написать код для построения встречного и прямого деревьев Фенвика, где F — функция, значение которой мы хотим считать на отрезках:↵

~~~~~↵
void bit_build(int n, int src[]) {↵
int m = 1; while(m < n) m *= 2; m -= 2;↵
for(int i = 0; i < n; ++i) {↵
a[i] = src[i]; // прямое дерево Фенвика↵
b[i] = i+1 < n ? src[i+1] : NEUTRAL; // встречное дерево Фенвика↵
}↵
for(int i = 0, j; i < n; ++i) {↵
const int j = i|i+1;↵
if(j < n) a[j] = F(a[i], a[j]);↵
}↵
for(int i = m-n+2; i < m; ++i) {↵
const int j = i|i+1;↵
const int u = m-i, v = m-j;↵
if(j < m) b[v] = F(b[v], b[u]);↵
}↵
}↵
~~~~~↵

При $n=2^k$ видно, что отрезки, покрываемые прямым и обратным деревом Фенвика идентичны дереву отрезков длины n. Из этого видно, что любой отрезок делится на два других отрезка так, что один из них полностью покрывается встречным деревом Фенвика, а другой &mdash; прямым.↵

Остается только сопоставить получившееся дерево Фенвика и ДО. Представлю иллюстрацию для случая n=8.↵

![ ](/predownloaded/23/88/238856306a0b1b7c471e139305c29e3671169e76.png)↵

Отсюда становятся очевидными три важных факта:↵

1. Каждая вершина ДО (кроме листьев) составленная из элементов дерева Фенвика имеет две дочерние вершины, имеющие одинаковые индексы, но одно из них лежит в прямом дереве Фенвике, а другое во встречном.↵

2. В двоичной системы счисления номера листья оканчиваются на "0", вершины лежащие на втором снизу уровне обладают суффиксом "01", на третьем &mdash; "011", четвертом &mdash; "0111" и т.д.↵

3. Номера вершин ДО, лежащих на одном уровне, строго возрастают в порядке слева-направо.↵

Рассматривая номера вершин на пути от листа $v$ легко проверить, что все вершины при их записи в двоичной системе счисления будут иметь префикс такой же, как у $v$, но суффикс будет другим, в зависимости от того, на каком уровне лежит вершина. Массив же, которому принадлежит вершина будем определять непосредственно смотря на значение очередного бита в номере листа. Пример пути от листа до верхней вершины дерева отрезков для позиции номер 20 в массиве.↵

![ ](/predownloaded/34/de/34debd82f52c4ceddd786fc8dffc93b25fd808f8.png)↵

Дальше, я надеюсь, идея обновления значения элемента в массиве становится понятной &mdash; просто обновляем значение листа, отвечающего за значение нужного элемента массива и идем вверх, пересчитывая значения всех рассматриваемых вершин. Поиск значения функции F на отрезке я описывать не буду, потому что (а) мне лень, и (б) алгоритм очень схож с поиском ответа на запрос в обычном дереве Фенвика, только в нашем случае мы дополнительно используем встречное дерево Фенвика.↵

~~~~~↵
// Обновление произвольного элемента в массиве↵
void bit_update(int id, int val) {↵
if(id&1) id=id^1, b[id] = r; // в зависимости от четности обновляем значение в нужном массиве↵
else              a[id] = r;↵
for(int j = ~1, t=2, z=1, p=id; ; z=z|t, t=t<<1) {↵
j = j^t; const int v = id&j|z; if(v >= n) break;↵
// При рассмотрении следующей вершины будем учитывать, что обе дочерние вершины↵
// имеют одинаковый индекс, т.е. находить их явно, нам не надо.↵
if(id&t) b[v] = F(a[p], b[p]);↵
else     a[v] = F(a[p], b[p]);↵
p = v;↵
}↵
}↵

// Ответ на запрос↵
int bit_ask(int l, int r) {↵
int res = NEUTRAL;↵
for(; (r&r+1) >= l && r >= l; r=(r&r+1)-1)↵
res = F(a[r], res);↵
for(; (l|l-1) <= r && l && l < n; l=(l|l-1)+1) ↵
res = F(b[l-1], res);↵
return res;↵
}↵
~~~~~↵

Код RMQ с использованием данной структуры:↵

<spoiler summary="Spoiler">↵
~~~~~↵
#include <stdio.h>↵

const int N = 1e5+7, MINF = -1e9;↵

int n, dx, m, a[N], b[N];↵

constexpr static int calc_dx(int n) {↵
int dx = 1, tmp = 0; while((tmp = dx << 1) < n) dx = tmp;↵
return dx;↵
}↵

int main() {↵
scanf("%d", &n); dx = calc_dx(n);↵
for(m = 1; m < n; m *= 2); m -= 2;↵
for(int i = 0; i < n; ++i) {↵
scanf("%d", &a[i]); if(i) b[i-1] = a[i];↵
} b[n-1] = MINF;↵

for(int i = 0, j; i < n; ++i) ↵
if((j=i|i+1) < n && a[j] < a[i]) a[j] = a[i];↵
for(int i = m-n+2; i < m; ++i) {↵
const int j = i|i+1;↵
const int u = m-i, v = m-j;↵
if(j < m && b[v] < b[u]) b[v] = b[u];↵
}↵

for(scanf("%d", &m); m--; ) {↵
char s[9]; int l, r;↵
scanf("%s%d%d", s, &l, &r);↵

if(s[0] == 'u') {↵
int id = l-1;↵
if(id&1) id=id^1, b[id] = r;↵
else              a[id] = r;↵
for(int j = ~1, t=2, z=1, p=id; ; z=z|t, t=t<<1) {↵
j = j^t; const int v = id&j|z; if(v >= n) break;↵
if(id&t) b[v] = a[p] > b[p] ? a[p] : b[p];↵
else     a[v] = a[p] > b[p] ? a[p] : b[p];↵
p = v;↵
}↵
}↵
else {↵
int res = MINF; --l; --r;↵
for(; (r&r+1) >= l && r >= l; r=(r&r+1)-1)↵
if(res < a[r]) res = a[r];↵
for(; (l|l-1) <= r && l && l < n; l=(l|l-1)+1) ↵
if(res < b[l-1]) res = b[l-1];↵
printf("%d ", res);↵
}↵
}↵
}↵
~~~~~↵


</spoiler>↵

P.S.: Сегодня протестировал свою программу на рандомных и не совсем рандомных тестах при различных ограничениях. По скорости сравнивал с [этой реализацией ДО](https://codeforces.me/blog/entry/18051). Оказалось, что ДО все же быстрее, при большом количестве запросов разница достигает 100мс. Выводы можно делать разные, например [user:Bugman,2019-08-16] предположил, что реализация встречного дерева Фенвика, где индексация будет начинаться не с нуля, а с единицы, может оказаться быстрее. Ну что ж, в ближайшее время постараюсь проверить.↵

### Таблица с результатами:↵

<table>↵
    <tr>↵
        <th>Ограничения</th>↵
        <th>Встречное дерево Фенвика</th>↵
        <th>Дерево отрезков</th>↵
        <th>Разница</th>↵
    </tr>↵
    <tr>↵
        <td>n=3e5, m=5e5</td>↵
        <td>156 ms</td><td>171 ms</td><td>+15 ms</td>↵
    </tr>↵
    <tr>↵
        <td>n=3e5, m=1e6</td>↵
        <td>265 ms</td><td>280 ms</td><td>+15 ms</td>↵
    </tr>↵
    <tr>↵
        <td>n=7e5, m=10e5</td>↵
        <td>343 ms</td><td>311 ms</td><td>-32 ms</td>↵
    </tr>↵
    <tr>↵
        <td>n=7e5, m=25e5</td>↵
        <td>780 ms</td><td>686 ms</td><td>-94 ms</td>↵
    </tr>↵
    <tr>↵
        <td>n=1e6, m=2e6</td>↵
        <td>686 ms</td><td>608 ms</td><td>-78 ms</td>↵
    </tr>↵
    <tr>↵
        <td>n=1e6, m=3e6</td>↵
        <td>872 ms</td><td>795 ms</td><td>-77 ms</td>↵
    </tr>↵
    <tr>↵
        <td>n=1e6, m=5e6</td>↵
        <td>1466 ms</td><td>1372 ms</td><td>-94 ms</td>↵
    </tr>↵
</table>

History

 
 
 
 
Revisions
 
 
  Rev. Lang. By When Δ Comment
ru5 Russian teraqqq 2019-08-16 17:52:03 1502 Мелкая правка: 'iler>\n\n**P.S.: ** Ребят, п' -> 'iler>\n\n*P.S.: * Ребят, п'
ru4 Russian teraqqq 2019-08-15 22:38:06 2
ru3 Russian teraqqq 2019-08-15 22:09:46 15
ru2 Russian teraqqq 2019-08-15 21:16:22 2884 (опубликовано)
ru1 Russian teraqqq 2019-08-15 20:47:32 4166 Первая редакция (сохранено в черновиках)