POJ 2785 4 Values whose Sum is 0(折半枚举)

题目链接

解题思路

  因为要从4列中各选1个数使和等于0，所以可以枚举任意两列和剩余的两列，那么前者的数量乘以后者中前者的相反数的数量就是本题答案。不过这个得用哈希但是好像会被卡的样子，那么可以把前者都存起来，然后枚举后者，通过二分计算数量。

代码

const int maxn = 4e3+10;
int n, k, a[maxn][4], b[maxn*maxn];
int main() {
    scanf("%d",&n);
    for (int i = 0; i<n; ++i)
        for (int j = 0; j<4; ++j)
            scanf("%d",&a[i][j]);
    for (int i = 0; i<n; ++i)
        for (int j = 0; j<n; ++j)
            b[k++] = a[i][0]+a[j][1];
    sort(b,b+k);
    ll ans = 0;
    for (int i = 0; i<n; ++i)
        for (int j = 0; j<n; ++j) 
            ans += upper_bound(b,b+k,-a[i][2]-a[j][3])-lower_bound(b,b+k,-a[i][2]-a[j][3]);
    printf("%lld
",ans);
    return 0;                                                                 
}