day2 POJ 2785 4 Values whose Sum is 0

#include"iostream"
#include"algorithm"
using namespace std;
int n,k,l,r,mid,mm,sum=0;
int a[4005],b[4005],c[4005],d[4005];
int p[4005*4005],q[4005*4005];
int main()
{
cin>>n;
for(int i=0;i<n;i++)
cin>>a[i]>>b[i]>>c[i]>>d[i];
k=0;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
p[k++]=a[i]+b[j];
k=0;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
q[k++]=c[i]+d[j];

sort(q,q+k);

for(int i=0;i<k;i++)
{
l=0; r=k-1;
while(l<=r)
{
mid=(l+r)/2;
if(q[mid]==-p[i]) break;
else if(q[mid]<-p[i]) l=mid+1;
else r=mid-1;
}
if(q[mid]==-p[i])
{
mm=mid;
while(q[mm]==-p[i]&&mm>=0) {sum++; mm--;}
mm=mid;
while(q[mm]==-p[i]&&mm<k) {sum++; mm++;}
sum--;
}

}
cout<<sum<<endl;
return 0;
}

给四个数组要求Ax+Bx+Cx+Dx=0的组数，将AB求和，CD求和，排序CD，之后枚举AB二分查找CD即可，因为要求解组数，而CD中可能存在多个相同项，所以每当在CD中查找成功的时候向左右两个方向遍历CD，如果有相同值那么sum++，别忘了遍历后sum结果-1，因为mid值计算了两次。