POJ 2785（4 Values whose Sum is 0）

#include<iostream>
#include<algorithm>
using namespace std;
int n,ans,a[4040],b[4040],c[4040],d[4040],ab[4040*4040],cd[4040*4040];
int main()
{
    int n;
        while(cin>>n)
        {
            for(int i=0;i<n;i++)
                cin>>a[i]>>b[i]>>c[i]>>d[i];
                int pab=0,pcd=0;
            for(int i=0;i<n;i++)
            {
                for(int j=0;j<n;j++)
                {
                    ab[pab++]=a[i]+b[j];
                    cd[pcd++]=-(c[i]+d[j]);
                }
            }
            sort(cd,cd+pcd);
            for(int i=0;i<pab;i++)
            {
               int mid,low=0,up=pcd-1;
               while(low<=up) 
               {
                   mid=(low+up)/2;
                   if(cd[mid]==ab[i])
                    {
                        ans++;
                        for(int j=mid+1;j<pcd;j++)
                        {
                            if(cd[j]==ab[i]) ans++;
                            else break;
                        }
                        for(int j=mid-1;j>=0;j--)
                        {
                            if(cd[j]==ab[i]) ans++;
                            else break;
                        }
                        break;
                    }
                    else
                    {
                       if(cd[mid]>ab[i]) up=mid-1;
                       else low=mid+1;
                    }
                   
               }
            }
            cout<<ans<<endl;
        }
        return 0;
}