程序设计思维与实践 Week4 作业 B 四个数列

题目描述：

ZJM 有四个数列 A,B,C,D，每个数列都有 n 个数字。ZJM 从每个数列中各取出一个数，他想知道有多少种方案使得 4 个数的和为 0。

当一个数列中有多个相同的数字的时候，把它们当做不同的数对待。

请你帮帮他吧！

Input：

第一行：n（代表数列中数字的个数） （1≤n≤4000）

接下来的 n 行中，第 i 行有四个数字，分别表示数列 A,B,C,D 中的第 i 个数字（数字不超过 2 的 28 次方）

Output：

输出不同组合的个数。

思路：

首先考虑最简单的做法，四个数列中取数，使得和为0，先想到的就是暴力枚举，时间复杂度为O(n^4)。

若将a+b+c+d=0看作是a+b=-(c+d)，继续枚举，AB数列和CD数列，将AB数列的和记录为数列x，再将CD数列的和记录为数列y，此时的时间复杂度为O(n^2)，然后可以去寻找对于xi，数列y中yi=-xi的个数，若y数列有序，则个数为（最后一个符合要求的yj坐标-第一个符合要求的yi坐标+1），因此通过枚举x序列，通过二分的方式寻找y数列中对应的答案。

#include<iostream>
#include<algorithm>
using namespace std;
int n,ans,a[4010],b[4010],c[4010],d[4010];
int sum,x[16000010],y[16000010];
int num,l,r,mid,pl,pr;
map<int,int> m;
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	cin>>a[i]>>b[i]>>c[i]>>d[i];
	for(int i=1;i<=n;i++)
	  for(int j=1;j<=n;j++)
	  {
	  	x[++sum]=a[i]+b[j];
	  	y[sum]=c[i]+d[j];
	  }
	sort(x+1,x+sum+1);
	sort(y+1,y+sum+1);
	for(int i=1;i<=sum;i++)
	{
		num=-x[i];
		l=1,r=sum;
		pl=0,pr=0;
		while(l<=r)
		{
			mid=(l+r)/2;
			if(y[mid]==num) pl=mid;
			if(y[mid]>=num) r=mid-1;
			else l=mid+1;
		}
		l=1,r=sum;
		while(l<=r)
		{
			mid=(l+r)/2;
			if(y[mid]==num) pr=mid;
			if(y[mid]<=num) l=mid+1;
			else r=mid-1;
		}
		if(pl!=0&&pr!=0)
			ans+=pr-pl+1;
	}
	cout<<ans;
	return 0;
}