POJ2785 Values whose Sum is 0

题意

给定四个整数值列表 A、B、C、D，计算有多少个四元组 (a, b, c, d ) ∈ A x B x C x D 使得 a + b + c + d = 0。四元组大小均为n。

思路

最简单的思路是$n^4$暴力，显然会TLE。
可以通过计算出ABC中三元组的和记为abc，在D中二分查找abc将算法复杂度降为$n^{3}logn$，依然会TLE。
依然运用二分思想，将AB的和保存下来，存入数组a中，再二重循环遍历CD计算出CD二元组的值cd，在a数组中查找cd出现的次数将其累加即可。

代码

#include <stdio.h>
#include <algorithm>
using namespace std;
int a[16000001],b[16000001];
int A[4001],B[4001],C[4001],D[4001];
int l,r,n,tot,mid,ans;
int find(int x)
{
	l=1,r=tot;
	while (l<=r)
	{
		mid=(l+r+1)>>1;
		if (a[mid]==x) return b[mid];
		if (a[mid]>x) r=mid-1;
		if (a[mid]<x) l=mid+1;
	}
	return 0;
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;++i) scanf("%d%d%d%d",&A[i],&B[i],&C[i],&D[i]);
	for (int i=1;i<=n;++i)
		for (int j=1;j<=n;++j) a[++tot]=A[i]+B[j];
	sort(a+1,a+1+tot);
	b[1]=1;
	int m;
	for (int i=2,j=1;i<=tot;++i)
		if (a[i]==a[i-1]) b[j]++;
		else b[++j]=1,a[j]=a[i],m=j;
	tot=m;
	for (int i=1;i<=n;++i)
		for (int j=1;j<=n;++j) ans+=find(0-C[i]-D[j]);
	printf("%d\n",ans);
	return 0;
}