题意:输入一个n,给出一个由n行4列组成的数组,要求在每一列找出一个数,使得四个数相加为0。输出一共有多少种情况。
题记:首先将四列的数变成两列,即将第一第二列的每两个数加起来存到一个新数组里,第三第四列一样如此。然后遍历第一第二列之和的数组,对于第一第二列之和的数组中每一个数都在第三第四列之和的数组中二分找出第一个大于等于这个数的负数的下标(lower_bound)然后判断这两个数加起来是否等于0即可。(注意会有多个加起来为0的情况,所以在找到这个数的下标之后要从这个下标往后遍历,直到两个数相加不为0)。
#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
const int N=4010,M=2e7;
int a[N][10];
int sum1[M];
int sum2[M];
int bsearch(int l,int r,int x){//手写二分
while(l<r){
int mid=l+r>>1;
if(sum2[mid]<x)l=mid+1;
else r=mid;
}
return l;
}
int main(){
int n;
cin>>n;
for(int i=1;i<=n;i++)
for(int j=1;j<=4;j++)
scanf("%d",&a[i][j]);
int k=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
sum1[k]=a[i][1]+a[j][2];
sum2[k]=a[i][3]+a[j][4];
k++;
}
}
sort(sum2,sum2+k);
long long ans=0;
for(int i=0;i<k;i++){
int pos=bsearch(0,k,-sum1[i]);
//int pos=lower_bound(sum2,sum2+k,-sum1[i])-sum2;
for(int j=pos;j<k;j++){
if(sum1[i]+sum2[j]==0) ans++;
else break;
}
}
cout<<ans<<endl;
return 0;
}