POJ 2785 4 Values whose Sum is 0 （二分）题解

最新推荐文章于 2019-02-11 19:59:14 发布

转载最新推荐文章于 2019-02-11 19:59:14 发布 · 77 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/KirinSB/p/9408794.html

本文介绍了一种解决四数求和问题的有效算法，通过将问题转化为前后两组数求和并利用二分查找的方式，从O(n^4)的时间复杂度降低到更优的级别。适用于需要高效处理大量数据的场景。

思路：

如果用朴素的方法算O（n^4）超时，这里用折半二分。把数组分成两块，分别计算前后两个的和，然后枚举第一个再二分查找第二个中是否有满足和为0的数。

注意和有重复

#include<iostream>
#include<algorithm>
#include<cstring>
#define ll long long
using namespace std;
const int N = 4000+5;
int a[N],b[N],c[N],d[N];
int mp1[N*N],mp2[N*N],cn1,cn2;
int main(){
    int n;
    scanf("%d",&n);
    for(int i = 1;i <= n;i++) scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
    cn1 = cn2 = 0;
    for(int i = 1;i <= n;i++){
        for(int j = 1;j <= n;j++){
            mp1[cn1++] = a[i] + b[j];
            mp2[cn2++] = c[i] + d[j];
        }
    }
    sort(mp2,mp2+cn2);
    ll ans = 0;
    for(int i = 0;i < cn1;i++){
        int x = lower_bound(mp2,mp2+cn2,-mp1[i]) - mp2;
        if(mp1[i] + mp2[x] == 0) ans += upper_bound(mp2,mp2+cn2,-mp1[i]) - mp2 - x;
    }
    printf("%lld\n",ans);
    return 0;
}

转载于:https://www.cnblogs.com/KirinSB/p/9408794.html