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：输出不同组合的个数。

解题思路：

当然最简单的办法是穷举法，依次枚举A，再枚举B，再枚举C，然后二分查找D是否存在，但是这样复杂度太高，是o（N^3logN)，所以必须优化该算法。本题思路是A+B=-(C+D），枚举A，B求得A+B的所有值，再枚举C，D求得C+D的所有值。对每一个A+B值，二分搜索求得C+D是否存在，若是存在，最小索引和最大索引分别是多少。

实验代码：

#include<iostream>
#include<algorithm>
using namespace std;
 
int a[5000];
int b[5000];
int c[5000];
int d[5000];
int ab[30000000];
int cd[30000000];

int binarySearch_min(int goal,int left,int right)		//找到最小序号 
{
	int l=left;
	int r=right;
	int ans=-1,mid;
	while(l<=r)				
	{
		mid=(l+r)/2;			
		if(cd[mid]<goal)		
			l=mid+1;
		else if(cd[mid]>goal)
			r=mid-1;
		else 					
			ans=mid;
			r=mid-1;
		} 
	}
	return ans;
}
int binarySearch_max(int goal,int left,int right)		//找到最大序号 
{
	int l=left;
	int r=right;
	int ans=-1,mid;
	while(l<=r)				
	{
		mid=(l+r)/2;			 
		if(cd[mid]<goal)		
			l=mid+1;
		else if(cd[mid]>goal)
			r=mid-1;
		else 				
		{
			ans=mid;
			l=mid+1;
		} 
	}
	return ans;	
} 
int getAns(int n)
{
	int size=0;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			ab[size]=a[i]+b[j];		
			cd[size]=c[i]+d[j];		
			size++;
		}
	}	
	//要对cd数组进行排序
	sort(cd,cd+size); 
	int ans=0;
	for(int i=0;i<size;i++)
	{
		int min=binarySearch_min(-ab[i],0,size-1);
		int max=binarySearch_max(-ab[i],0,size-1);
		if(min!=-1)		
			ans+=max-min+1;
	}
	return ans;
}
int main(void)
{
	int n;		//表示每个数组有n个数据 
	cin>>n;
	for(int i=0;i<n;i++)
		cin>>a[i]>>b[i]>>c[i]>>d[i];
	cout<<getAns(n)<<endl;
	return 0;
}