4 Values whose Sum is 0(二分法)

题目链接:http://poj.org/problem?id=2785

题意:给你4个数组a,b,c,d,问从每一个数组中任选一个元素,如果这4个元素的和为0,则说这4个数为quadruplet,问总共有多少组quadruplet。

题解:纯枚举的话,时间复杂度为o(n^4),肯定会超时,所以用二分的思想做这道题。我们想到既然有四列,那么我们可以分别计算前两列和后两列的和(只需要n*n*2次运算) 然后对后两列的和排序,那么我们对于每一个前两列的和都可以二分找到后两列的和中与之相加为0的个数,这样的复杂度就是O(n*n*log(n))是可以接受的。(来自neko1)

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 5000;
const int maxm = 25000000;
long long a[maxn], b[maxn], c[maxn], d[maxn];
long long sum2[maxm];

int main()
{
    int n;
    long long num;
    while (scanf ("%d", &n) != EOF) {
        num = 0;
        for (int i = 0; i < n; i++) {
            scanf ("%lld%lld%lld%lld", &a[i], &b[i], &c[i], &d[i]);
        }
        int cnt = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                sum2[cnt++] = c[i] + d[j];
            }
        }
        sort (sum2, sum2 + cnt);
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                long long sum = a[i] + b[j];
                num += upper_bound(sum2, sum2 + cnt, -sum) - lower_bound(sum2, sum2 + cnt, -sum);
            }
        }
        printf ("%lld\n", num);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值