eolymp
bolt
Спробуйте наш новий інтерфейс для відправки розв'язків
Задачі

Делящийся на 3

Делящийся на 3

Для массива целых чисел $[b_1, b_2, ..., b_m]$ определим его вес как сумму попарных произведений его элементов, а именно как сумму $b_i \cdot b_j$ по всем $1 \le i < j \le m$. Задан массив $n$ целых чисел $[a_1, a_2, ..., a_n]$. Найдите количество пар чисел $(l, r)$ таких что $1 \le l \le r \le n$, для которых вес подмассива $[a_l, a_{l+1}, ..., a_r]$ делится на $3$. \InputFile Первая строка содержит одно число $n~(1 \le n \le 5 \cdot 10^5)$ --- длину массива. Вторая строка содержит $n$ целых чисел $a_1, a_2, ..., a_n~(0 \le a_i \le 10^9)$ --- элементы массива. \OutputFile Выведите одно целое число --- количество пар целых чисел $(l, r)$, где $1 \le l \le r \le n$, для которых вес соответствующего подмассива делится на $3$.
Ліміт часу 1 секунда
Ліміт використання пам'яті 128 MiB
Вхідні дані #1
3
5 23 2021
Вихідні дані #1
4
Вхідні дані #2
5
0 0 1 3 3
Вихідні дані #2
15
Вхідні дані #3
10
0 1 2 3 4 5 6 7 8 9
Вихідні дані #3
20
Джерело 2020 SEERC South Eastern European Regional Programming Contest, Винница и Бухарест, Май 23, Задача E