Codeforces и Polygon могут быть недоступны в период с 23 мая, 7:00 (МСК) по 23 мая, 11:00 (МСК) в связи с проведением технических работ. ×

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

Автор MAX11189, история, 23 месяца назад, По-русски

Дано два массива одинаковой длины, нужно с помощью прекрасных массивов найти любой подмассив, где сумма елементов первого равна сумме елементов второго, решить надо за O(n), заранее спасибо за помощь) если вдруг кто-то поможет :)

  • Проголосовать: нравится
  • 0
  • Проголосовать: не нравится

»
23 месяца назад, # |
Rev. 2   Проголосовать: нравится 0 Проголосовать: не нравится

Если я правильно понял, условие звучит так: даны массивы a и b длиной n, найти любые 0 <= l <= r < n, такие, что a[l] + ... + a[r] = b[l] + ... + b[r].

В голову пришло такое решение: давайте посчитаем массив префиксных сумм для обоих массивов: prefa[i] = a[0] + ... + a[i], аналогично для prefb и b. Теперь, посчитаем разностный массив для этих двух массивов, т.е. diff[i] = prefa[i] — prefb[i].

В таком случае, если нужные l и r существуют, то diff[l — 1] = diff[r]. Значит, проверим, что в массиве diff какое-то значение встречается хотя бы два раза, если это так — можем найти ответ.

К сожалению, поскольку значения в diff могут находиться в очень большом диапазоне, я не знаю способ, как решать задачу за честную линию, но с помощью unordered_map можно решать за амортизированную линию. Надеюсь, поможет :)