【POJ】2785 - 4 Values whose Sum is 0 折半查找

http://poj.org/problem?id=2785

《挑战程序设计竞赛》 P160

有n个整数的四个数列A,B,C,D。每个数列取一个数，使得4个数之和为0.

折半查找。对半成AB和CD在考虑，使得c+d=-(a+b)。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
#include <algorithm>
using namespace std;

int n;
int A[4005],B[4005],C[4005],D[4005];
int CD[4005*4005];

int main(){

    cin >> n;
    for (int i=0;i<n;i++){
        cin >> A[i] >> B[i] >> C[i] >> D[i];
    }

    for (int i=0;i<n;i++){
        for (int j=0;j<n;j++){
            CD[i*n+j]=C[i]+D[j];
        }
    }
    sort(CD,CD+n*n);

    long long ans=0;
    for (int i=0;i<n;i++){
        for (int j=0;j<n;j++){
            int cd=-(A[i]+B[j]);
            //取出Ｃ和Ｄ中和为ｃｄ的部分 
            ans+=upper_bound(CD,CD+n*n,cd)-lower_bound(CD,CD+n*n,cd);  //数组中cd的个数
        }
    }

    cout << ans << endl;
}