题解 二分基础 POJ2785

题意:给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++;

posted on 2014-07-16 10:43  一锅土豆  阅读(119)  评论(0编辑  收藏  举报