UVA1152：4 Values whose Sum is 0（和为0的4个值）

作者：xq的acm之路

题目链接：https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=3593

题目大意：给定4个n元素集合A,B,C,D，要求从中分别选出一个元素a,b,c,d,使得a+b+c+d=0，问有多少种写法。

思路：刚开始用不定数组vector存储A+B，然后C+D，扫描数组里有多少个元素符合，发现超时，就想起了ub和lb这两个函数，

假设数组int point【5】={1，2，7，7，9},ans=7;

upper_bound使用方法：

int tem=upper_bound（point,point+5，ans）-point //ans为待查找元素，输出为ans最大可以插入数组point里面的位置

输出：4；

lower_bound使用方法：

int tem=lower_bound（point,point+5，ans）-point //ans为待查找元素，输出为ans最小可以插入数组point里面的位置

输出：2；

因此，使用ub和lb可以查询数组里某个元素的个数，比扫描数组一遍要省时得多。

当然，对不定数组使用这两个函数要用到迭代器。

至于迭代器，可以百度，，，

代码如下：

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main()
{
    int n;
    cin>>n;
    while(n--)
    {
        vector<int>ab;
        int a[4001],b[4001],c[4001],d[4001],cd[4001];
        int t,sum=0;
        cin>>t;
        for(int i=0; i<t; i++)
        {
            cin>>a[i]>>b[i]>>c[i]>>d[i];
        }
        for(int i=0; i<t; i++)
        {
            for(int j=0; j<t; j++)
            {
                int m=a[i]+b[j];
                ab.push_back(m);
            }
        }
        sort(ab.begin(),ab.end());
        for(int i=0; i<t; i++)
        {
            for(int j=0; j<t; j++)
            {
                int ans=(c[i]+d[j])*(-1);
                int x=lower_bound(ab.begin(),ab.end(),ans)-ab.begin();
                int y=upper_bound(ab.begin(),ab.end(),ans)-ab.begin();
                sum+=y-x;
            }
        }
        cout<<sum<<endl;
        if(n>0) cout<<endl;
    }
    return 0;
}