四个数列 Week4作业B题

题目：

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

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

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

输出：
输出不同组合的个数。

样例：
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

思路：

有四个数列，暴力枚举的复杂度为O(n^4)，不可取。
运用分而治之的思想，将A和B、C和D他们分别求和看作是两个数组，即求对应的相反数，可将复杂度降至O(n^2)。
枚举C和D的时候计算它的相反数在A和B中出现多少次，也就是计算一个数在有序数列中第一次和最后一次出现的位置。

代码：

#include <iostream>
#include <algorithm>
#include <map>
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;
}