1046. 4个值的和为0

单点时限: 15.0 sec

内存限制: 256 MB

这个 ” 和 ” 的问题，可以被这样定义 : 给定 4 个列表的整数 A,B,C,D, 计算有多少个四元组 (a,b,c,d)∈AxBxCxD, 使得 a+b+c+d=0. 我们假定所有列表的大小为 n.

输入格式
输入的第一行是列表的大小 n(不超过 4000). 我们接下去将给出 n 行，每行有 4 个整数 (每个的绝对值不超过 2^28), 代表 A,B,C,D.

输出格式
你的程序需要输出一个整数，表示有多少个四元组的和为 0.

Hint:

Sample Explanation: Indeed, the sum of the five following quadruplets is zero: (-45, -27, 42, 30), (26, 30, -10, -46), (-32, 22, 56, -46),(-32, 30, -75, 77), (-32, -54, 56, 30).

样例
input
6
-45 22 42 -16
-41 -27 56 30
-36 53 -37 77
-36 30 -75 -46
26 -38 -10 62
-32 -54 -6 45
output
5

/*
upper_bound(first, last, val) 寻找在数组或容器的[first,last)范围第一个大于val的元素位置
lower_bound(first, last, val)寻找在数组或容器的[first,last)范围第一个大于等于val的元素位置
使用时必须为有序的数组或容器
*/
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=4000;
int n;
int A[maxn],B[maxn],C[maxn],D[maxn];
int AB[maxn*maxn],CD[maxn*maxn];
void solve() {
	for(int i=0; i<n; ++i)
		for(int j=0; j<n; ++j) {
			AB[i*n+j]=A[i]+B[j];
			CD[i*n+j]=C[i]+D[j];
		}
	sort(AB,AB+n*n);
	sort(CD,CD+n*n);
	long long ans=0;
	for(int i = 0; i < n*n; i++) {
		int cd=-AB[i];
		ans+=upper_bound(CD, CD+n*n, cd) - lower_bound(CD, CD+n*n, cd);
	}
	printf("%lld\n",ans);
}

int main() {
	scanf("%d",&n);
	for(int i=0; i<n; ++i) {
		scanf("%d%d%d%d",&A[i],&B[i],&C[i],&D[i]);
	}
	solve();
	return 0;
}