给四个有n个元素的集合ABCD,各挑一个数abcd使得abcd等于0,问有几种挑法。
先枚举c+d存到数组里 然后枚举a+b,看有多少c+d=-(a+b)。
统计的时候利用了upper_bound和lower_bound的区别:当存在时,两个的差值就是a+b+c+d=0的个数;当不存在时,两个值相等(没有等于只有大于的数),差值就为0。
有推荐使用哈希表的,很遗憾我不会。
再吐槽一句UVA的格式要求实在太那啥了,一定要多输出一个换行干啥呢
AC代码:
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn=4010;
int a[maxn],b[maxn],c[maxn],d[maxn];
int map[maxn*maxn],map2[maxn*maxn];
int main()
{
int n;
int t;
cin>>t;
while(t--)
{
cin>>n;
for(int i=0;i<n;i++)
scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
map[i*n+j]=c[i]+d[j];
sort(map,map+n*n);
long long ans=0;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
{
int cd=-(a[i]+b[j]);
ans+=upper_bound(map,map+n*n,cd)-lower_bound(map,map+n*n,cd);
}
cout<<ans<<endl;
if(t) printf("\n");
}
return 0;
}