题目地址:UVa 1152
先枚举A集合与B集合的和,存起来,然后再枚举C集合与D集合的和,看与存起来的值有多少个是互为相反数的。水题。找存起来的值时可以用二分找。
代码如下:
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include <set>
#include <algorithm>
using namespace std;
int a[5000], b[5000], c[6000], d[6000], p[17000000];
int main()
{
int t, n, m, ans, i, j, k;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
m=0;
for(i=0;i<n;i++)
{
scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
}
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
{
p[m++]=a[i]+b[j];
}
}
sort(p,p+m);
int low, high, mid;
ans=0;
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
{
int x=-c[i]-d[j];
low=0, high=m-1;
while(low<=high)
{
mid=low+high>>1;
if(p[mid]>x) high=mid-1;
else if(p[mid]<x) low=mid+1;
else
{
for(k=mid;k>=0;k--)
{
if(p[k]==x)
{
ans++;
}
else
break;
}
for(k=mid+1;k<m;k++)
{
if(p[k]==x)
{
ans++;
}
else
break;
}
break;
}
}
}
}
printf("%d\n",ans);
if(t)
puts("");
}
return 0;
}