Блог пользователя I_love_myself

Автор I_love_myself, 6 лет назад, По-русски

Всем привет, хочу с вами поделиться интересной задачей и классной идеей.

Необходимо реализовать структуру данных, выполняющих 2 типа запросов:
- Изменить элемент в массиве
- Найти сумму различных элементов с L по R.
Например массив выглядит так: { 1, 2, 1 }, тогда сумма различных от 1 до 3 равна 3

Для начала разберем "лобовое" решение задачи за с помощью корневой оптимизации. Спасибо Holidin, который объяснил мне это решение.

Построим персистентное ДО для суммы различных элементов на отрезке. Заведем еще одно ДО с unordered_map<int, int> в вершине, которое хранит количество элементов, равных X на отрезке.
Ещё будем накапливать до изменений, если из стало , то перестроим наше персистентное ДО заново за . Чтобы ответить на запрос суммы на отрезке, посчитаем ее с помощью первого ДО, затем пройдемся по всем накопленным изменениям и проверим, входят ли они в наш отрезок и как они меняют сумму с помощью второго ДО с map'ами. Ответ на запрос будет за .

Теперь решим задачу нормально за в онлайн.

Для начала пусть нам надо просто считать количество различных элементов на отрезке с изменениями.

Ключевая идея: пусть для каждого элемента мы знаем next[i] — индекс следующей позиции j, такой, что a[i] = a[j], i < j. Тогда количество различных элементов на отрезке [L; R] равно количеству элементов на том же отрезке, у которых next[i] больше R.

Тогда решение очень простое: построим ДО, в котором можно будет считать количество элементов на отрезке [L; R], больших R, для этого в вершине будем хранить ДД по ключу(next[i]). Чтобы обновлять next[i], я лично поддерживал map<int, set>, элемент которого m[x] означает множество позиций, на которых стоят x.

Теперь вернемся к нашей задаче. Чтобы добить её, нужно в декартаче по ключу(next[i]) хранить сумму a[i] в поддереве.

UPD: Ссылка на задачу Спасибо kefaa и khadaev

Решение в онлайн
  • Проголосовать: нравится
  • +32
  • Проголосовать: не нравится

»
6 лет назад, # |
  Проголосовать: нравится +11 Проголосовать: не нравится
»
6 лет назад, # |
Rev. 4   Проголосовать: нравится +3 Проголосовать: не нравится

Могу предложить еще одну интересную задачу. Дан массив a состоящий из n натуральных чисел, не превосходящих n. Ответить на q запросов двух типов:

  1. set p v — выполнить присвоение a[p] = v
  2. get l r — выдать сумму таких элементов на отрезке [l, r], которые встречаются чаще остальных.

Ограничения: 2 секунды, 64 МБ, 1 ≤ n, q ≤ 2·105, 1 ≤ l ≤ r ≤ n, 1 ≤ p, v, a[i] ≤ n.

Пример:

5 3
1 2 3 2 3
get 1 5 // ответ 5, частота элементов: p[1] = 1/5, p[2] = 2/5, p[3] = 2/5, поэтому 2 + 3 = 5
set 1 2
get 1 5 // ответ 2, частота элементов: p[2] = 3/5, p[3] = 2/5, поэтому 2
»
4 года назад, # |
Rev. 3   Проголосовать: нравится -10 Проголосовать: не нравится

Я переписал этот код на яву (точь в точь вплоть до имплементации) и получаю TL. На некоторых тестах код работает больше 10с. Этот код на плюсах вообще проходит? Потому что 10^5 запросов * log ДО * 4log ДД * 6 * немалую скрытую константу.