POJ 2785 4 Values whose Sum is 0(雙向搜索+二分)

題意：給4個數組，從每個數組中選一個數，求出4個數和爲0的方案數。

分析：暴力時間複雜度爲N^3,肯定不行。所以考慮先把ab、cd的和分別求出來。-(a+b)=c+d即滿足條件，求和複雜度爲N*N。ab數組爲-(a+b)，cd數組爲(c+d)。

從cd數組裏找等於ab數組的即可。這裏可以先給數組排序 ，然後用二分搜索找。複雜度爲O(N*N*logN)。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<cmath>
#define ll __int64
#define INF 0x3fffffff
#define M 4005
using namespace std;

int n;
int a[M],b[M],c[M],d[M];
int ab[M*M],cd[M*M];

int main()
{
    //freopen("d:\\Test.txt","r",stdin);
    while(cin>>n){
        for(int i=0;i<n;i++){
            scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
        }
        int k1=0;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                ab[k1++]=-a[i]-b[j];
            }
        }
        int k2=0;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                cd[k2++]=c[i]+d[j];
            }
        }
        sort(ab,ab+k1);
        sort(cd,cd+k2);
        ll ans=0;
        for(int i=0;i<k1;i++){
            ans+=upper_bound(cd,cd+k2,ab[i])-lower_bound(cd,cd+k2,ab[i]);
        }
        cout<<ans<<endl;
    }
    return 0;
}