题目描述
给出3个整数集合A B C,每个数据集合中有n个元素,集合中的数有正有负且各不相等,现在分别从A B C中各选出一个元素a b c,使得a+b+c=0。请问一共有多少种组合?
输入格式
第一行,一个整数n表示每个集合中元素的数量n(n≤2000)
接下来n行,每行是用空格隔开的3个整数(-2^28<a,b,c<2^28),分别表示数据集合A B C中的值
输出格式
输出一行一个整数,表示使得a+b+c=0的组合数
输入输出样例
输入样例1:
5 -3 2 5 3 -2 1 0 -4 -1 6 7 -5 10 9 -8
输出样例1:
7
说明
一共有7种组合,分别是:
-3 2 1
-3 -2 5
3 2 -5
3 -2 -1
3 -4 1
6 2 -8
10 -2 -8
【耗时限制】1000ms 【内存限制】128MB
#include<bits/stdc++.h>
# define N 2010
using namespace std;
int n,a[N],b[N],c[N];
int low(int l,int r,int x){
while(l<r){
int mid=(l+r)/2;
if(c[mid]>=x){
r=mid;
}
else {
l=mid+1;
}
}
return l;
}
int up(int l,int r,int x){
while(l<r){
int mid=(l+r)/2;
if(c[mid]>x){
r=mid;
}
else {
l=mid+1;
}
}
return l;
}
int main(){
cin>>n;
for(int i=1;i<=n;i++){
scanf("%d %d %d",&a[i],&b[i],&c[i]);
}
int cnt=0;
sort(c+1,c+n+1);
sort(a+1,a+n+1);
sort(b+1,b+n+1);
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
int x=0-a[i]-b[j];
cnt+=up(1,n+1,x)-low(1,n+1,x);
}
}
cout<<cnt;
return 0;
}