二分 D 和为0

1、简单描述

有四列数据，共n行，每列数据找一个，这四个数加起来为0，问一共有多少种方法使和为0？

2、思路

   

#include <stdio.h>  
    #include <iostream>  
    #include <string.h>  
    #include <algorithm>  
    using namespace std;  
    int a[4001][4],sum1[20000002],sum2[20000002];  
    int main()  
    {  
        int n,mid;  
        cin>>n;
            for(int i=0; i<n; i++)  
            {  
                scanf("%d%d%d%d",&a[i][0],&a[i][1], &a[i][2],&a[i][3]);  
            }  
            int k=0;  
            int m=0;  
            for(int i=0; i<n; i++)  
                for(int j=0; j<n; j++)  
                {  
                    sum1[k++]=a[i][0]+a[j][1];  
                    sum2[m++]=a[i][2]+a[j][3];  
                }  
            sort(sum2,sum2+k);  
            int cnt=0;  
            for(int i=0; i<k; i++)  
            {  
                int left=0;  
                int right=k-1;  
                while(left<=right)  
                {  
                    mid=(left+right)/2;  
                    if(sum1[i]+sum2[mid]==0)  
                    {  
                        cnt++;  
                        for(int j=mid+1;j<k;j++)  
                        {  
                            if(sum1[i]+sum2[j]!=0)  
                               break;  
                            else  
                            cnt++;  
                        }  
                        for(int j=mid-1;j>=0;j--)  
                        {  
                            if(sum1[i]+sum2[j]!=0)  
                               break;  
                            else  
                               cnt++;  
                        }                                                           //在mid左右查找元素，看是否有与mid相等的元素。若有，++
                        break;  
                    }  
                    if(sum1[i]+sum2[mid]<0)  
                        left=mid+1;  
                    else  
                        right=mid-1;  
                }  
            }  
            cout<<cnt<<endl; 
        return 0;  
    }  

输入数据，第一列元素分别与第二列元素相加，结果存在sum1中，第三列元素与第四列元素分别相加，结果存在sum2中；对sum2排序，对于sum1中的每一个数查找，用二分查找。