配点 : 600 点
問題文
長さ N の整数列 a= {a1,a2,⋯,aN} と、整数 K が与えられます。
a の空でない連続する部分列 {al,al+1,⋯,ar} (1≤l≤r≤N) は N(N+1)/2 個存在します。これらのうち、算術平均が K 以上であるものは何個あるでしょうか?
制約
- 入力値はすべて整数である。
- 1≤N≤2×105
- 1≤K≤109
- 1≤ai≤109
入力
入力は以下の形式で標準入力から与えられる。
N K
a1
a2
:
aN
出力
a の空でない連続する部分列のうち、算術平均が K 以上であるものの個数を出力せよ。
3 6
7
5
7
5
以下に、a のすべての空でない連続する部分列を示します。
- {a1} = {7}
- {a1,a2} = {7,5}
- {a1,a2,a3} = {7,5,7}
- {a2} = {5}
- {a2,a3} = {5,7}
- {a3} = {7}
これらの平均はそれぞれ 7, 6, 19/3, 5, 6, 7 であり、このうち 6 以上であるものは 5 個です。なお、{a1} と {a3} は含まれる要素の値では区別できませんが、これらは個別に数えます。
1 2
1
0
7 26
10
20
30
40
30
20
10
13