#5168. Problem 2. Farmer John Solves 3SUM

0

Problem 2. Farmer John Solves 3SUM

Problem 2. Farmer John Solves 3SUM

USACO 2020 January Contest, Gold

Farmer John 相信他在算法设计上实现了一个重大突破:他声称他发现了一个 3SUM 问题的近似线性时间算法,这是一个有名的算法问题,尚未发现比运行速度比平方时间明显更优的解法。3SUM 问题的一个形式是:给定一个整数数组 s1,,sms_1,\dots,s_m,计算不同索引组成的无序三元对 i,j,ki,j,k 的数量,使得 si+sj+sk=0s_i + s_j + s_k = 0

为了测试 Farmer John 的断言,Bessie 提供了一个 NN 个整数组成的数组 AA1N50001 \leq N \leq 5000)。Bessie 还会进行 QQ 次询问(1Q1051 \leq Q \leq 10^5),每个询问由两个索引 1aibiN1 \leq a_i \leq b_i \leq N 组成。对于每个询问,Farmer John 必须在子数组 A[aibi]A[a_i \dots b_i] 上求解 3SUM 问题。

不幸的是,Farmer John 刚刚发现了他的算法中的一个错误。他很自信他能修复这个算法,但同时,他请你帮他先通过 Bessie 的测试!

测试点性质: 测试点 2-4 满足 N500N\le 500。测试点 5-7 满足 N2000N\le 2000。测试点 8-15 没有额外限制。

输入格式(文件名:threesum.in):

输入的第一行包含两个空格分隔的整数 NNQQ。第二行包含空格分隔的数组 AA 的元素 A1,,ANA_1,\dots,A_N。以下 QQ 行每行包含两个空格分隔的整数 aia_ibib_i,表示一个询问。

保证对于每个数组元素 AiA_i106Ai106-10^6 \leq A_i \leq 10^6

输出格式(文件名:threesum.out):

输出包含 QQ 行,第 ii 行包含一个整数,为第 ii 个询问的结果。注意你需要使用 64 位整数型来避免溢出。

输入样例:


7 3
2 0 -1 1 -2 3 3
1 5
2 4
1 7

输出样例:


2
1
4

对于第一个询问,所有的三元对为 (A1,A2,A5)(A_1,A_2,A_5)(A2,A3,A4)(A_2,A_3,A_4)

供题:Dhruv Rohatgi