题目链接:http://poj.org/problem?id=2785
题目的大意是:从四列中分别选取一个数使得总和为0,问一共有多少种取法。
注意如果第一列有两个0,用第一个0和第二个0是两种不同的取法。
感觉直接做肯定会超时所以没有去暴力。上网查了解题报告说是哈希。
第一次接触哈希,看了好多资料(有点浮躁,总是看不进去)。最后终于写出来了。
下面是几个注意点 :首先哈希会存在冲突,所以有冲突的解决方法。有很多,这里用的是另外找一个地址。具体看代码。
其中key的取值的技巧是一个大大素数。
代码不难理解,一开始老是出错,原来是那个hash要定义为全局变量。因为在main函数里面的话,数组开不出来。
时间效率不是很高,这个应该和数组的大小有关系。数组越大的话冲突越少效率也就越高把。
下面是代码:
#include<cstdio>
using namespace std;
int const HMAX = 3000000;
int const KEY = 1357;
const int INF = 1000000000;
struct Hash{
private:
int data[HMAX];
int mark[HMAX];
public :
Hash(){
for(int i = 0; i < HMAX; ++i){
mark[i] = 0; data[i] = INF;
}
}
void Insert(int x){
int temp = x;
temp = (temp + INF) % HMAX;
while(data[temp] != INF && data[temp] != x)
temp = (temp + KEY) % HMAX;
data[temp] = x;
++mark[temp];
}
int find(int x){
int temp = x;
temp = (temp + INF) % HMAX;
while(data[temp] != INF && data[temp] != x)
temp = (temp + KEY) % HMAX;
if(data[temp] == INF) return 0;
else if(data[temp] == x) return mark[temp];
}
};
Hash hash;
int main()
{
int a[4005][2], b[4005][2];
int i, j, k;
int n;
scanf("%d", &n);
for(i = 0; i < n; ++i)
scanf("%d%d%d%d", a[i], a[i] + 1, b[i], b[i] + 1);
for(i = 0; i < n; ++i)
for(j = 0; j < n; ++j)
hash.Insert(a[i][0] + a[j][1]);
k = 0;
for(i = 0; i < n; ++i)
for(j = 0; j < n; ++j)
k += hash.find((-1) * (b[i][0] + b[j][1]));
printf("%d\n", k);
return 0;
}