K12859 和为0的三个数

题目描述

给出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;
}