Контест Middle Слайсы и мапы изнутри обновлено

Число подмассивов с суммой k на Go

Условие

Посчитайте, сколько непрерывных подмассивов имеют сумму ровно k.

Ввод: n k (1 ≤ n ≤ 1.5·10⁴, |k| ≤ 10⁹), затем n целых чисел, |aᵢ| ≤ 10⁴. Числа могут быть отрицательными.

Вывод: одно число.

Пример

5 3
1 2 1 2 1
4

Как подойти

Префиксные суммы и мапа частот: ответ += cnt[p - k], затем cnt[p]++.

Решите в тренажёре

Редактор с автотестами прямо в браузере. Можно решить спокойно, вслух или под таймером, как на собесе.