关闭

UVA1152:4 Values whose Sum is 0(和为0的4个值)

标签: uva
178人阅读 评论(0) 收藏 举报
分类:

作者: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;
}
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:8645次
    • 积分:700
    • 等级:
    • 排名:千里之外
    • 原创:62篇
    • 转载:2篇
    • 译文:0篇
    • 评论:3条