UVA-1152 4 Values whose Sum is 0 (二分)

时间:2023-03-10 04:27:26
UVA-1152 4 Values whose Sum is 0 (二分)

题目大意:在4个都有n个元素的集合中,每个集合选出一个元素,使得4个数和为0。问有几种方案。

题目分析:二分。任选两组求和,剩下两组求和,枚举第一组中每一个和sum,在第二组和中查找-sum的个数,累加起来便得答案。

代码如下:

# include<iostream>
# include<cstdio>
# include<vector>
# include<cstring>
# include<algorithm>
using namespace std; int a[4000][4];
vector<int>v1,v2; int f1(int l,int r,int x)///查找下界
{
int lp,m;
while(l<r){
m=l+(r-l)/2;
if(v2[m]<x){
lp=l=m+1;
}else
lp=r=m;
}
return lp;
} int f2(int l,int r,int x)///查找上界
{
int up,m;
while(l<r){
m=l+(r-l)/2;
if(v2[m]<=x)
up=l=m+1;
else
up=r=m;
}
return up;
} int main()
{
int T,n;
scanf("%d\n",&T);
while(T--)
{
scanf("%d",&n);
v1.clear(),v2.clear();
for(int i=0;i<n;++i)
for(int j=0;j<4;++j)
scanf("%d",&a[i][j]);
for(int i=0;i<n;++i){
for(int j=0;j<n;++j){
v1.push_back(a[i][0]+a[j][1]);
v2.push_back(a[i][2]+a[j][3]);
}
} int ans=0;
sort(v2.begin(),v2.end());
for(int i=0;i<n*n;++i){
int lpos=f1(0,n*n,-v1[i]);
int upos=f2(0,n*n,-v1[i]);
ans=ans+upos-lpos;
}
printf("%d\n",ans);
if(T)
printf("\n");
}
return 0;
}