#ARC170B. Arithmetic Progression Subsequence

Arithmetic Progression Subsequence

Arithmetic Progression Subsequence

时间限制:2s
空间限制:1024MB

题目描述

给定一个长度为 NN、元素均在 11 到 1010 之间的整数序列 AA。

称一对整数 (l,r)(l,r) 为好区间,当且仅当 1≤l≤r≤N1\le l\le r\le N,并且 Al,Al+1,…,ArA_l,A_{l+1},\ldots,A_r 含有一个长度为 33 的等差子序列。也就是说,存在 l≤i<j<k≤rl\le i<j<k\le r,满足

Aj−Ai=Ak−Aj.A_j-A_i=A_k-A_j.

求好区间的数量。

输入格式

第一行输入整数 NN。

第二行输入 NN 个整数 A1,A2,…,ANA_1,A_2,\ldots,A_N。

输出格式

输出好区间的数量。

样例输入 1

5
5 3 4 1 5

样例输出 1

3

样例输入 2

3
1 2 1

样例输出 2

0

样例输入 3

9
10 10 1 3 3 7 2 2 5

样例输出 3

3

数据范围

  • 3≤N≤1053\le N\le 10^5;
  • 1≤Ai≤101\le A_i\le 10;
  • 输入中的所有数均为整数。
子任务编号 分值 特殊限制
1 15 所有 AiA_i 相同
2 20 N≤20N\le 20
3 35 N≤2000N\le 2000
4 30 无特殊限制