四个数列（二分）

题意：

ZJM 有四个数列 A,B,C,D，每个数列都有 n 个数字。ZJM 从每个数列中各取出一个数，他想知道有多少种方案使得 4 个数的和为 0。
当一个数列中有多个相同的数字的时候，把它们当做不同的数对待。
请你帮帮他吧！

输入输出说明：

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

输出不同组合的个数。

样例输入：

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

样例输出：

5

样例解释：

5组数据如下： (-45, -27, 42, 30), (26, 30, -10, -46), (-32, 22, 56, -46),(-32, 30, -75, 77), (-32, -54, 56, 30)

思路：

暴力做法：
枚举ABCD中的每个数，四重循环，时间复杂度𝑂(𝑛^4) ,超时，考虑如何减少时间复杂度。

优化：
A+B+C+D=0可以变形为A+B=-(C+D)，当然也可以变形为A=-(B+C+D)等其他形式，这里我们变形为A+B=-(C+D)，这样我们就将一个四重循环优化成一个双重循环。

二分：
如何确定满足条件的组合个数？依次取出A+B中的数值，查找C+D中满足C+D=-(A+B）的个数有几个。那么问题又来了，如何找到C+D中满足条件的个数？将C+D排序，只要有序就可以使用二分算法，二分算法就是每次将查找范围缩小一半，降低了时间复杂度。利用二分算法，求满足要求的数字第一次出现的位置l和最后一次出现的位置r，那么，r-l+1就是这个数字出现的次数。

总结：

1.考虑如何将四次方的算法进行优化：算式变形
2.二分查找的妙用：只要有序，都可以使用二分
3.数组开大亿点点，1≤n≤4000，那么A+B的数组大小就可能到达16000000，因为数组开小了wa了好几次==

代码：

#include<iostream>
#include<algorithm>
using namespace std;

int n,p=0,res=0;
int a[4001],b[4001],c[4001],d[4001];
int r1[16000001],r2[16000001];
bool cmp(const int& o1,const int& o2){return o1<o2;}

int findFirst(int x){
	int l=0,r=p-1,ans=-1;
	while(l<=r){
		int mid=(l+r)>>1;
		if(r2[mid]==x){
			ans=mid;  r=mid-1;
		}
		else if(r2[mid]>x) r=mid-1;
		else l=mid+1;
	}
	return ans;
}

int findLast(int x){
	int l=0,r=p-1,ans=-1;
	while(l<=r){
		int mid=(l+r)>>1;
		if(r2[mid]==x){
			ans=mid;  l=mid+1;
		}
		else if(r2[mid]>x)  r=mid-1;
		else l=mid+1;
	}
	return ans;
}

int main(){
	cin>>n;
	for(int i=0;i<n;i++)
		cin>>a[i]>>b[i]>>c[i]>>d[i];
	for(int i=0;i<n;i++){
		int k1=a[i]; int k2=c[i]; 
		for(int j=0;j<n;j++){
			int t1=b[j];  int t2=d[j];
			r1[p]=k1+t1;  r2[p]=k2+t2;  p++;
		}
	}
	//sort(r1,r1+p,cmp);
	sort(r2,r2+p,cmp);
	for(int i=0;i<p;i++){
		int y=r1[i]*(-1);
		int tt=findLast(y),kk=findFirst(y);
		if(kk!=-1)
			res+=(tt-kk+1);
	}
	cout<<res<<endl;
	return 0;
}