AT_arc170_b [ARC170B] Arithmetic Progression Subsequence

给定一个由 $1$ 到 $10$ 之间的整数构成的长度为 $N$ 的数列 $A$。 对于满足 $1\leq l \leq r \leq N$ 的整数对 $(l, r)$，如果满足以下条件，则称其为“良好组”： - 数列 $(A_l, A_{l+1}, \ldots, A_r)$ 至少包含一个长度为 $3$ 的等差数列作为（不一定连续的）子序列。更严格地说，存在整数组 $(i, j, k)$ 满足 $l \leq i < j < k \leq r$，且 $A_j - A_i = A_k - A_j$。 请计算良好组的个数。

输入通过标准输入按以下格式给出： > $N$ $A_1$ $\ldots$ $A_N$

请输出答案。

## 限制条件 - $3 \leq N \leq 10^5$ - $1 \leq A_i \leq 10$ - 输入的所有数均为整数 ## 样例解释 1 良好组有 $(l, r) = (1, 4), (1, 5), (2, 5)$ 共 $3$ 个。例如，数列 $(A_1, A_2, A_3, A_4)$ 包含了 $(5, 3, 1)$ 这样一个长度为 $3$ 的等差数列作为子序列，因此 $(1, 4)$ 是良好组。 ## 样例解释 2 也有可能不存在良好组。 由 ChatGPT 4.1 翻译