Хорошие подмассивы

Дан массив $a_1, \ldots, a_n$ из цифр от $0$ до $9$. Подмассив $a_l, \ldots, a_r$ называется хорошим, если сумма его элементов равна его длине: $\sum_{i=l}^{r} a_i = r - l + 1$. Найдите количество хороших подмассивов.

В первой строке $n$ ($1 \le n \le 10^5$). Во второй — строка из $n$ цифр (без пробелов).

Одно число — количество хороших подмассивов.

Вычтите $1$ из каждого элемента: $b_i = a_i - 1$. Теперь условие — $$, то есть два префикса равны: . Посчитайте, сколько раз встречается каждое значение префикса — значение, встретившееся раз, даёт пар. Префиксы лежат в , поэтому вместо словаря хватит массива со сдвигом.