POJ 2785 4 Values whose Sum is 0

链接：http://poj.org/problem?id=2785

题意：一组4个数，给出若干组，输出有多少对组合满足四个数之和为0

将四列数并成两列，对两列二分查找

#include <cstdio>
#include <algorithm>
using namespace std;
int t[4010][4];
int first2[4000*4000+4],last2[4000*4000+4];
int Binsearch(int total,int key) {
    int l=0,r=total-1;
    bool flag=0;
    while(l<=r) {
        int mid=(l+r)>>1;
        if(first2[mid]==key) {
            flag=1;
            break;
        } else if(first2[mid]>key) r=mid-1;
        else l=mid+1;
    }
    if(flag) {
        int sum;
        for(sum=0; l<total; l++) {
            if(first2[l]==key) sum++;
            if(sum && first2[l]!=key) break;
        }
        return sum;
    }
    return 0;
}
int main() {
    int n;
    while(~scanf("%d",&n)) {
        int k=0;
        for(int i=0; i<n; i++) for(int j=0; j<4; j++) scanf("%d",&t[i][j]);
        for(int i=0; i<n; i++)
            for(int j=0; j<n; j++) {
                first2[k]=t[i][0]+t[j][1];
                last2[k++]=t[i][2]+t[j][3];
            }
        sort(first2,first2+k);
        sort(last2,last2+k);
        int ans=0;
        for(int i=0; i<k; i++) ans+=Binsearch(k,-last2[i]);
        printf("%d\n",ans);
    }
}