一、题目大意

从四个数组A[ ],B[ ],C[ ],D[ ]中分别取一个元素a b c d，使得 a + b + c + d = 0，找出所有a b c d 的解的数量，认为下标不同，但值相同的元素为不同元素。

二、解题思路

如果暴力枚举，一定超时，所以我们把A[i]+B[i]的所有n*n种排列放在ab数组中，把C[i]+D[i]的所有n*n中排列放在cd数组中，然后把cd数组排序，二分从cd数组中查找-ab[i]的所有值即可。

二分时可以使用0x3f3f3f3f给cd数组第n*n+1个元素赋值，便于二分兜底。如果lower_bound查出的值与-ab[i]相等，则将upper_bound-lower_bound的值加给result

最后输出result即可，题目比较简单，在《挑战程序设计》上也可以找到，就不再过多的赘述了。

三、代码

#include <iostream>
#include <algorithm>
using namespace std;
int a[4007], b[4007], c[4007], d[4007], ab[16000007], cd[16000007], n, inf = 0x3f3f3f3f;
void input()
{for (int i = 0; i < n; i++){scanf("%d%d%d%d", &a[i], &b[i], &c[i], &d[i]);}
}
void preHandle()
{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(cd, cd + n * n);cd[n * n] = inf;
}
void solve()
{int result = 0;for (int i = 0; i < n * n; i++){int abSum = ab[i];int cdSum = 0 - abSum;int startIdx = lower_bound(cd, cd + n * n + 1, cdSum) - cd;int endIdx = upper_bound(cd, cd + n * n + 1, cdSum) - cd;if (cd[startIdx] != cdSum){continue;}result += (endIdx - startIdx);}printf("%d\n", result);
}
int main()
{while (~scanf("%d", &n)){input();preHandle();solve();}return 0;
}