题解二分基础 POJ2785

最新推荐文章于 2024-02-28 14:04:49 发布

weixin_30767921

最新推荐文章于 2024-02-28 14:04:49 发布

阅读量78

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/ghh1995/p/4349036.html

本文介绍了一种解决四个数组中寻找元素组合使其和为零的问题的方法。通过将两个数组的元素组合相加并使用二分查找进行匹配，提高了搜索效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：给A,B,C,D四组长度相同为n的数组，在每个数组中分别找一个数a,b,c,d使得 a+b+c+d=0。问一共有多少组这样的数据？

做法：将A和B左右数组排列相加的所有情况并放入vector1中，C和D的也按上述方式放入vector2中。在v1中用二分查找所有和v2中元素相加为0的数。

代码：

#include <iostream>
#include <vector>
#include <algorithm>
#include <iterator>
using namespace std;
int n;
vector <int> va,vb,vc,vd,v1,v2,v3;

int main()
{
    int n;
    int x1,x2,x3,x4,tot=0;
    int i,j;
    cin>>n;
    for(i=0;i<n;i++)
    {
        cin>>x1>>x2>>x3>>x4;
        va.push_back(x1),vb.push_back(x2),vc.push_back(x3),vd.push_back(x4);
    }
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
                v1.push_back(va[i]+vb[j]);
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
                v2.push_back(vc[i]+vd[j]);
        sort(v1.begin(),v1.end());
        sort(v2.begin(),v2.end());
       for(i=0;i<n*n;i++)
       {
           if(binary_search(v2.begin(),v2.end(),-v1[i]))
            tot=tot+upper_bound(v2.begin(),v2.end(),-v1[i])-lower_bound(v2.begin(),v2.end(),-v1[i]);
       }
        cout<<tot<<endl;

    return 0;
}

错误原因：1 超时用枚举法会超时；

2 WA 每次tot增加，必须是

upper_bound(v2.begin(),v2.end(),-v1[i])-lower_bound(v2.begin(),v2.end(),-v1[i])

而不是 if（bound_search(<span style="font-family: 'Microsoft YaHei';">v2.begin(),v2.end(),-v1[i])</span>)）tot++;

转载于:https://www.cnblogs.com/ghh1995/p/4349036.html

题解 二分基础 POJ2785

题解二分基础 POJ2785