POJ 2785 4 Values whose Sum is 0

#include <iostream>
#include <algorithm>
using namespace std;


int a[4001],b[4001],c[4001],d[4001];

int cd[4001*4001];

int main()
{
    int n;
    while(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[n*i+j] = c[i]+d[j];
            }
        }
        //排序 
        sort(cd,cd+n*n);
        
        long long res = 0;
        
        //枚举第二部分 
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                int ab = -(a[i]+b[j]);
                //二分思想查找值 
                res += upper_bound(cd,cd+n*n,ab) - lower_bound(cd,cd+n*n,ab);
            }
        }
        
        
        cout<<res<<endl;
        
    }
    return 0;
}