题目描述
四个数列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)复杂度显然是不能接受的,而简单变换形式之后,就可以使用二分的思想进行求解,大大优化了程序运行的时间。