#P10740. [SEERC 2020] Divisible by 3
[SEERC 2020] Divisible by 3
题目描述
定义一个序列 的权重为 。
现在你有一个长度为 的数组 ,求一共存在多少种 使得 且 的权重能被 整除。
输入格式
第一行一个整数 。
然后 个整数 。
输出格式
输出方案总数。
提示
对于第一个样例,存在 、、、 共 种方案。
定义一个序列 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)。
输出方案总数。
对于第一个样例,存在 [1,1]、[2,2]、[3,3]、[1,3] 共 4 种方案。