#L4140. 「PA 2024」Bardzo Ulubiony Ciąg

    ID: 4380 传统题 1000ms 256MiB 尝试: 2 已通过: 1 难度: 10 上传者: 标签>数据结构树状数组前缀和组合数学组合计数三元组问题枚举优化哈希表双指针PA2024

「PA 2024」Bardzo Ulubiony Ciąg

题目描述

题目译自 PA 2024 Runda 5 Bardzo Ulubiony Ciąg,感谢 Macaronlin 提供翻译。

给定长度为 nn 的整数数组 aaaa 的所有子区间和形成长度为 n(n+1)2\frac{n(n+1)}{2} 的数组 bb,子区间和按区间开始下标的递增顺序排列,如果区间开始下标相同,则按区间结束递增顺序排列。

对于新形成的数组 bb,求满足 bi+bj+bk=0 (i<j<k)b_i + b_j + b_k = 0\ (i < j < k)(i,j,k)(i,j,k) 对数。

输入格式

第一行一个整数 n (1n500)n\ (1 \le n \le 500),表示数组 aa 的长度。

第二行 nn 个整数 a1,a2,,an (ai20,000)a_1, a_2, \ldots, a_n\ (|a_i| \le 20,000),表示数组 aa

输出格式

一行一个整数,表示数组 bb 中满足 bi+bj+bk=0 (i<j<k)b_i + b_j + b_k = 0\ (i < j < k)(i,j,k)(i,j,k) 对数。 样例1:

3
7 -4 -2
1

数组 bb[7,3,1,4,6,2][7, 3, 1, -4, -6, -2],只有 3,1,43, 1, -4 这三个不同元素满足,所以答案为 11

样例2:

10
0 0 0 0 0 0 0 0 0 0
26235

数组 bb555500,任选三个 i<j<ki < j < k 都满足和为 00,因此答案为 26,23526,235

数据规模与约定

对于 100%100\% 的数据,0n1070 \le n \le 10^7