UVA 1152 4 Values whose Sum is 0

分析

题目不难理解，但是有一点要注意。题目的意思是在1、2、3、4每一位上的数字可以改变顺序，但是1、2、3、4位之间不能改变顺序。

首先想简单点，如果用枚举来做，n^4一定会超时的。但是如果我们换一种思路，把这四个数字分成两部分，1/2为一组，3/4为一组，然后来二分，这样一下从n^4降到n^2。

程序

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
using namespace std;
const int MAXN = 4000 + 1;
int n, a[MAXN], b[MAXN], c[MAXN], d[MAXN], ab[MAXN*MAXN], Count;
int main()
{
    cin >> n;
    for (int i = 1; i <= n; i++)
        cin >> a[i] >> b[i] >> c[i] >> d[i];
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= n; j++)
            ab[++Count] = a[i] + b[j];
    sort(ab+1, ab+(Count+1));
    int ans = 0;
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            ans += upper_bound(ab+1, ab+Count+1, -c[i]-d[j])-lower_bound(ab+1, ab+Count+1, -c[i]-d[j]);
        }
    }
    cout << ans << endl;
    return 0;
}