UVA - 1152 4 Values whose Sum is 0

题目大意：在4个都有n个元素的集合中，每个集合选出一个元素，使得4个数和为0。问有几种方案。

题目分析：开始想到暴力枚举，对四组数枚举，复杂度O(n^4)无法接受，后来对前三组数枚举累加，在第四组中找和的个数，如果对 第四组数据排序并二分查找，算法复杂度为O(n^3 logn)无法接受；最后采用二分。任选两组求和，剩下两组求和，枚举第一 组中 每一个和sum，在第二组和中查找-sum的个数，累加起来便得答案。

第一次用map映射前两组的sum，超时：

#include <cstdio>
#include <map>
using namespace std;
const int maxn = 4000 + 10;
int qua[4][maxn];

int main()
{
	int T, first = 1;
	scanf("%d", &T);
	while(T--) {
		int n;
		scanf("%d", &n);
		for(int i=0; i<n; i++) {
			for(int j=0; j<4; j++) {
				scanf("%d", &qua[j][i]);
			}
		}
		map<int, int> mp;
		for(int i=0; i<n; i++) {
			for(int j=0; j<n; j++) {
				mp[qua[0][i]+qua[1][j]]++;
			}
		}
        int cnt = 0;
        for(int i=0; i<n; i++) {
            for(int j=0; j<n; j++) {
                cnt += mp[-qua[2][i]-qua[3][j]];
            }
        }
        if(first)first=0;
        else putchar('\n');
        printf("%d\n", cnt);
	}
    return 0;
}

第二次直接用数组存放sum，循环外sort，查找边界即可：

#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 4000 + 10;
int qua[4][maxn];
int sum[maxn*maxn];

int main()
{
	int T, first = 1;
	scanf("%d", &T);
	while(T--) {
		int n;
		scanf("%d", &n);
		for(int i=0; i<n; i++) {
			for(int j=0; j<4; j++) {
				scanf("%d", &qua[j][i]);
			}
		}
		int c = 0;
		for(int i=0; i<n; i++) {
			for(int j=0; j<n; j++) {
				sum[c++] = qua[0][i]+qua[1][j];
			}
		}
		sort(sum, sum+c);
        int cnt = 0;
        for(int i=0; i<n; i++) {
            for(int j=0; j<n; j++) {
				int t = -qua[2][i]-qua[3][j];
                cnt += upper_bound(sum, sum+c, t) - lower_bound(sum, sum+c, t);
            }
        }
        if(first)first=0;
        else putchar('\n');
        printf("%d\n", cnt);
	}
    return 0;
}