【二分】四个数列

问题描述

ZJM 有四个数列 A,B,C,D，每个数列都有 n 个数字。ZJM 从每个数列中各取出一个数，他想知道有多少种方案使得 4 个数的和为 0。
当一个数列中有多个相同的数字的时候，把它们当做不同的数对待。
请你帮帮他吧！

Input

第一行：n（代表数列中数字的个数，1≤n≤4000）
接下来的 n 行中，第 i 行有四个数字，分别表示数列 A,B,C,D 中的第 i 个数字（数字不超过 2 的 28 次方）

Output

输出不同组合的个数。

Sample

Input

6
-45 22 42 -16
-41 -27 56 30
-36 53 -37 77
-36 30 -75 -46
26 -38 -10 62
-32 -54 -6 45

Output

5

思路

这是一道十分有趣的题目，直接爆搜n⁴的复杂度显然是过不去的，但是本题允许n²的复杂度，注意一共有四个数列，所以我们可以考虑拆成两部分，中间通过某种策略去搜索，显然我们可以通过有序二分的方式降低复杂度。先对前两个数列枚举求和排序，再枚举后两个数列的和，二分该和的相反数的左右区间，求差加一后累加，即可得到结果。

总结

二分时注意，要考虑边界的情况，根据二分策略对ans的初值进行调整，这样才可以保证二分出来的结果为满足该条件的第一个位置。

代码

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iomanip>
#include<vector>
using namespace std;
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define mem(a,s) memset(a,s,sizeof(a))
int a[4][4001];
int n;
vector<int> v;
int f1(int x){//>=的第一个位置，可能不是x ,若>=不存在，返回0 
 int l=0,r=n*n-1,ans=n*n;
 while(l<=r){
  int m=(l+r)>>1;
  if(v[m]>=x){
   ans=m;
   r=m-1;
  }else l=m+1;
 }
 return ans;
}
int f2(int x){//>x的第一个位置，若>不存在，返回0（边界点搜不到） 
 int l=0,r=n*n-1,ans=n*n;
 while(l<=r){
  int m=(l+r)>>1;
  if(v[m]>x){
      ans=m;
   r=m-1;
  }else l=m+1;
 }
 return ans;
}
int main(){
// freopen("1.txt","r",stdin);
 ios::sync_with_stdio(false);
 cin>>n;
 rep(i,1,n)
  cin>>a[0][i]>>a[1][i]>>a[2][i]>>a[3][i];
 rep(i,1,n)
  rep(j,1,n)
   v.push_back(a[2][i]+a[3][j]);
 sort(v.begin(),v.end());
 int sum=0;
 rep(i,1,n)
  rep(j,1,n){
  int x=-(a[0][i]+a[1][j]);
  int l=f1(x);
  int r=f2(x);
  //cout<<x<<' '<<l<<' '<<r<<' '<<l-r<<endl;
  sum+=(r-l);
 } 
 cout<<sum;
 return 0;
}