atcoder#ABC248H. [ABC248Ex] Beautiful Subsequences

[ABC248Ex] Beautiful Subsequences

配点 : 600600

問題文

(1,,N)(1,\ldots,N) を並び替えて得られる長さ NN の順列 P=(P1,,PN)P=(P_1,\ldots,P_N)、及び整数 KK が与えられます。

以下の条件を全て満たす整数組 (L,R)(L,R) の個数を求めてください。

  • 1LRN1 \leq L \leq R \leq N
  • $\mathrm{max}(P_L,\ldots,P_R) - \mathrm{min}(P_L,\ldots,P_R) \leq R - L + K$

制約

  • 1N1.4×1051 \leq N \leq 1.4\times 10^5
  • PP(1,,N)(1,\ldots,N) を並び替えて得られる順列
  • 0K30 \leq K \leq 3
  • 入力は全て整数

入力

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

NN KK

P1P_1 P2P_2 \ldots PNP_N

出力

答えを出力せよ。

4 1
1 4 2 3
9

条件を満たす組 (L,R)(L,R) は以下の 99 個です。

  • (1,1)(1,1)
  • (1,3)(1,3)
  • (1,4)(1,4)
  • (2,2)(2,2)
  • (2,3)(2,3)
  • (2,4)(2,4)
  • (3,3)(3,3)
  • (3,4)(3,4)
  • (4,4)(4,4)

(L,R)=(1,2)(L,R) = (1,2) は $\mathrm{max}(A_1,A_2) -\mathrm{min}(A_1,A_2) = 4-1 = 3$ 、RL+K=21+1=2R-L+K=2-1+1 = 2 となるので、条件を満たしません。

2 0
2 1
3
10 3
3 7 10 1 9 5 4 8 6 2
37