单点时限: 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;
}