题目
给你4个长度为N的数组,从每个数组中拿出一个数,加起来为0的情况有几种。从不同位置拿出来的就叫不同情况。
题解思路
将问题简化为A + B = -(C+D) ,枚举C+D的所有和进行排序,对于枚举出的A+B二分查找相反值。
时间复杂度化简为N平方log N
当原数组为升序时,upper_bound查找第一个大于值的数,lower_bound查找第一个大于等于的数。作差就是相等的值。
AC代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
#include <map>
#include <string>
using namespace std;
const int INF = 0x3f3f3f3f;
const int N = 4010 ;
int a[N],b[N],c[N],d[N];
int sum[N*N];
int main ()
{
ios::sync_with_stdio(false);
int n;
cin>>n;
for (int i = 1 ; i <= n ; i++ )
{
cin>>a[i]>>b[i]>>c[i]>>d[i];
}
for ( int i = 1 ; i <= n ; i++ )
{
for ( int k = 1 ; k <= n ; k++ )
{
sum[(i-1)*n + k ] = c[i] + d[k] ;
}
}
sort(sum+1,sum+1+n*n);
int ans = 0 ;
for ( int i = 1 ; i <= n ; i++ )
{
for ( int k = 1 ; k <= n ; k++ )
{
int cd = -(a[i]+b[k]) ;
ans += upper_bound(sum+1,sum+1+n*n,cd) - lower_bound(sum+1,sum+1+n*n,cd);
}
}
cout<<ans<<"\n";
return 0 ;
}