程序设计Week4——B-四个数列

题目描述

四个数列A、B、C、D，每个数列n个数字，从四个数列中各取一个数，求共有多少种方案使得4个数和为0。当一个数列中有多个相同的数字的时候，把它们当做不同的数对待。

Input

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

Output

输出不同组合的个数。

解题思路

暴力枚举必然不可取，n<=4000的四重循环必然超时，考虑O（n^2）的复杂度算法，因为A+B+C+D=0相当于A+B=-(C+D)，可以使用二分。
首先枚举AB数列的所有和，再枚举CD数列的和，在枚举CD和时，通过二分找到其相反数在AB数列和中第一次出现和最后一次出现的位置，由此可以统计出每组A+B=-(C+D)个个数，最后输出总个数即可。

实现代码

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

int a[4010],b[4010],c[4010],d[4010];
int ab_sum[16000010];

int find_first(int x,int l,int r)
{
	int ans=-1;
	while(l<=r)
	{
		int mid=(r+l)/2;
		if(x==ab_sum[mid])
		{//找x出现的第一个位置，故继续向左查找 
			r=mid-1;
			ans=mid;
		}
		else if(x>ab_sum[mid])
			l=mid+1;
		else
			r=mid-1;
	}
	return ans;
}

int find_last(int x,int l,int r)
{
	int ans=-1;
	while(l<=r)
	{
		int mid=(r+l)/2;
		if(x==ab_sum[mid])
		{//找x出现的最后一个位置，故继续向右查找 
			l=mid+1;
			ans=mid;
		}
		else if(x>ab_sum[mid])
			l=mid+1;
		else
			r=mid-1;
	}
	return ans;
}

int main()
{
	int n,m=-1,cd_sum=0,total=0;
	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)
		{
			m++;
			ab_sum[m]=a[i]+b[j];
		}
	}
	sort(ab_sum,ab_sum+m+1);
	for(int i=0;i<n;++i)
	{
		for(int j=0;j<n;++j)
		{
			cd_sum=-(c[i]+d[j]);
			if(find_first(cd_sum,0,m)!=-1)
				total+=(find_last(cd_sum,0,m)-find_first(cd_sum,0,m)+1);
		}
	}
	cout<<total<<endl;
	return 0;
} 

总结

本题是一道二分的题目，利用二分可以找到一组数中第一个（最后一个）比给定数字大或小的数，利用这种形式，我们可以很方便的找到满足某种条件的数字或表达式的个数，带入到本题当中，常规的四重循环O（n^4）复杂度显然是不能接受的，而简单变换形式之后，就可以使用二分的思想进行求解，大大优化了程序运行的时间。