UVA1152 4 Values whose Sum is 0

题目大意：给定4个n(1 <= n <= 4000)元素集合A, B, C, D，要求分别从中选取一个元素a, b, c, d，使得a+b+c+d = 0，问有多少种选法。

时间限制：9000ms

通过时间：1983ms

分析：

最容易想到的是四重循环枚举a, b, c, d，看看加起来是否是0，时间复杂度O(n^4)，超时。

这时我们采用一种叫做“中途相遇法”的算法，先枚举a, b，把所有a+b的值记录下来，然后枚举c, d，查一查-c-d有多少种方法写成a+b的形式，总时间复杂度O(n^2logn)

那么，问题就是如何记录呢？

不难想到STL里有个东西叫map，但不幸还是超时。

这里推荐的高效算法是哈希，hd, nxt不用介绍，w[i]表示第i个结点存储的数（也就是a+b），st[i]表示第i个结点有多少种表示方法，使用哈希需要注意这么几个问题：

1.hd数组初始化成0，所以结点编号应从1开始。

2.因为插入查询的数可能是负数，所以应采用(x % hashsize + hashsize) % hashsize来得到它的哈希编号。

最终答案可能爆int，注意用long long。

最后就是多组数据记得清空。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int hashsize = 1000003;
int n, t, cnt, a[4005], b[4005], c[4005], d[4005], hd[hashsize], nxt[16000005], w[16000005], st[16000005];
long long ans;

void in(int x) {
	int h = (x % hashsize + hashsize) % hashsize, u = hd[h];
	while(u) {
		if(w[u] == x) {
			st[u]++;
			return;
		}
		u = nxt[u];
	}
	nxt[++cnt] = hd[h];
	hd[h] = cnt;
	w[cnt] = x;
	st[cnt] = 1;
}

int srch(int x) {
	int h = (x % hashsize + hashsize) % hashsize, u = hd[h];
	while(u) {
		if(w[u] == x) return st[u];
		u = nxt[u];
	}
	return 0;
}

int main() {
	scanf("%d", &t);
	while(t--) {
		cnt = ans = 0;
		scanf("%d", &n);
		memset(hd, 0, sizeof hd);
		for(int i = 1; i <= n; i++)
			scanf("%d%d%d%d", &a[i], &b[i], &c[i], &d[i]);
		for(int i = 1; i <= n; i++)
		for(int j = 1; j <= n; j++)
			in(a[i]+b[j]);
		for(int i = 1; i <= n; i++)
		for(int j = 1; j <= n; j++)
			ans += srch(-c[i]-d[j]);
		printf("%lld\n", ans);
		if(t) printf("\n");
	}
	return 0;
}