【折半枚举】 poj 2875

poj2785 四个数的和

题意：给出四个数组，要求每个数组中选出一个数，四个数的和为0，问总共有几种选法。数组元素个数<=4000

n^4的复杂度降到n^2logn

#include<iostream>
#include<algorithm>
using namespace std;
const int N=4e3+10;
int a[N],b[N],c[N],d[N];
int half[N*N];
int n;
int cnt=0;
int main()
{
    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 j=1;j<=n;++j)
            half[cnt++]=c[i]+d[j];
    sort(half,half+cnt);
    int ans=0;
    for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j)
        {
            int x=a[i]+b[j];
            x=-x;
            ans+=upper_bound(half,half+cnt,x)-half-(lower_bound(half,half+cnt,x)-half);
        }
    cout<<ans<<endl;


}