poj 2785 4 Values whose Sum is 0

这个题目有点思想。给出四列数字 要找有多少组合使得四个数字和为0 这四个数字分别来自这四个列

暴力肯定是T了

怎么能减少时间呢？

我们用另外两个数组 一个存第一列和第二列的和 一共有n*n个元素

另一个存第三列和第四列的和 一共n*n个元素

然后排序第二个数组

对第一个数字中每一个数 二分查找第二个数组看有没有相反数

如果找到了 还要再第二个数组前后探查一下 有没有相等的

 

 

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=4005;
int arr1[maxn];
int arr2[maxn];
int arr3[maxn];
int arr4[maxn];
int arr5[maxn*maxn];
int arr6[maxn*maxn];
int n,i,j,k;
void solve()
{
    int counter=0;
    for(i=0; i<k; ++i)
    {
        int mid,low=0,high=k-1;
        bool f=0;
        while(low<=high)
        {
            mid=(low+high)/2;
            if(arr6[mid]==arr5[i])
            {
                f=1; break;
            }
            else if(arr5[i]<arr6[mid])
                high=mid-1;
            else
                low=mid+1;
        }
        if(f==1)
        {
            counter++;
            for(j=mid-1; j>=0; j--)
            {
                if(arr6[j]==arr5[i])
                    counter++;
                else break;
            }
            for(j=mid+1; j<k; ++j)
            {
                if(arr6[j]==arr5[i])
                    counter++;
                else break;
            }
        }
    }
    printf("%d\n",counter);
}

int main()
{
    while(~scanf("%d",&n))
    {
        for(i=1; i<=n; ++i)
            scanf("%d%d%d%d",&arr1[i],&arr2[i],&arr3[i],&arr4[i]);
        k=0;
        for(i=1; i<=n; ++i)
        {
            for(j=1; j<=n; ++j)
            {
                arr5[k]=arr1[i]+arr2[j];
                arr6[k++]=-(arr3[i]+arr4[j]);
            }
        }
        sort(arr6,arr6+k);
        solve();
    }
    return 0;
}