100 #ABC146E. [ABC146E] Rem of Sum is Num

[ABC146E] Rem of Sum is Num

题目描述

長さ N N の正整数列 A1, A2,  , AN A_1,\ A_2,\ \ldots\ ,\ A_N と正の整数 K K が与えられます。

A A の空でない連続する部分列であって、要素の和を K K で割った余りが要素の数と等しくなるものの数を求めてください。ただし、2 2 つの部分列が列として同じでも、取り出す位置が異なるならば区別するものとします。

输入格式

入力は以下の形式で標準入力から与えられる。

N N K K A1 A_1 A2 A_2 \cdots AN A_N

输出格式

条件をみたす部分列の個数を出力してください。

题目大意

给定一个序列和一个 KK,求这个序列有多少个子区间(子串)满足该区间内所有元素和 SS 除以 KK 的余数是区间长度。

即存在多少对 (l,r)(l,r),满足 0<rl+1<K0<r-l+1<KK  ((al+al+1++ar)(rl+1))K\ |\ ((a_l+a_{l+1}+\cdots+a_r)-(r-l+1))

5 4
1 4 2 3 5
4
8 4
4 2 4 2 4 2 4 2
7
10 7
14 15 92 65 35 89 79 32 38 46
8

提示

制約

  • 入力は全て整数である。
  • 1  N  2× 105 1\ \leq\ N\ \leq\ 2\times\ 10^5
  • 1  K  109 1\ \leq\ K\ \leq\ 10^9
  • 1  Ai  109 1\ \leq\ A_i\ \leq\ 10^9

Sample Explanation 1

(1) (1) , (4,2) (4,2) , (1,4,2) (1,4,2) , (5) (5) 4 4 つが条件をみたす部分列です。

Sample Explanation 2

(4,2) (4,2) 4 4 回、(2,4) (2,4) 3 3 回数えられています。