四个数组取数为0 （二分策略）

问题描述

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

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

请你帮帮他吧！

Input

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

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

Output

输出不同组合的个数。

Sample 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

Sample Output

5

解题思路以及关键代码

第一反应暴力枚举，但是注意数据范围，n<=4000的范围肯定不能四重循环来枚举。
我们采用先枚举A和B，得到A和B的和ab数组，然后枚举C和D，在枚举C和D的时候，计算它的相反数在ab数组中出现的次数。这样时间复杂度就可以控制在O(n^2)

for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			tmp=-(c[i]+d[j]);
			if(find2(tmp,size)!=-1)
			answer=answer+find2(tmp,size)-find1(tmp,size)+1;		
		}
	}

全部代码

#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 4000;
int a[maxn];
int b[maxn];
int c[maxn];
int d[maxn];
int s[maxn*maxn];
int answer=0;
int find1(int x,int n)  //找到x出现的第一个位置
{
	int l=0,r=n-1,ans=-1;
	while(l<=r)
	{
		int mid=(l+r)/2;
		if(s[mid]==x)
		{
			ans = mid;
			r = mid-1;
		}
		else if(s[mid] > x) r= mid-1;
		else l=mid+1;
	}
	return ans;
}
int find2(int x,int n) //找到x出现的最后一个位置
{
	int l=0,r=n-1,ans=-1;
	while(l<=r)
	{
		int mid=(l+r)/2;
		if(s[mid]==x)
		{
			ans = mid;
			l = mid+1;
		}
		else if(s[mid] > x) r= mid-1;
		else l=mid+1;
	}
	return ans;
}

int main()
{
	int n;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++)
	{
		for(int j=0;j<n;j++)
		{
			s[i*n+j]=a[i]+b[j];
		}
	}
	int size = n*n;
	sort(s,s+size);
	int tmp;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			tmp=-(c[i]+d[j]);
			if(find2(tmp,size)!=-1)
			answer=answer+find2(tmp,size)-find1(tmp,size)+1;		
		}
	}
	cout<<answer; 
	return 0;
}