题目描述
定义一个序列 b 的权重为 ∑i=1n∑j=1i−1bi×bj。
现在你有一个长度为 n 的数组 a,求一共存在多少种 (l,r) 使得 1≤l≤r≤n 且 [al,al+1,…,ar] 的权重能被 3 整除。
输入格式
第一行一个整数 n (1≤n≤5×105)。
然后 n 个整数 ai (0≤ai≤109)。
输出格式
输出方案总数。
3
5 23 2021
4
5
0 0 1 3 3
15
10
0 1 2 3 4 5 6 7 8 9
20
提示
对于第一个样例,存在 [1,1]、[2,2]、[3,3]、[1,3] 共 4 种方案。