poj 2785 hash

题意：

给出4个数组，每个数组里面挑一个数，和为0；

 

分析：

把前两个数组加起来，hash，枚举后两个数组加起来 的相反数

 

注意：multiset会超时；手写hash

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <set>
#include <vector>

using namespace std;

#define MAX 1000000000
#define size 20345677
#define key 745

int hash[size],sum[size];

void hash_insert(int num) {
    int tmp = num;
    num = (num + MAX)%size;
    while(hash[num]!=MAX&&hash[num]!=tmp) {
        num = (num + key)%size;
    }
    hash[num] = tmp;
    sum[num]++;
}

int hash_Find(int num) {
    int tmp = num;
    num = (num + MAX)%size;
    while(hash[num]!=MAX&&hash[num]!=tmp) {
        num = (num + key)%size;
    }
    if(hash[num]==MAX) return 0;
    return sum[num];
}

int a[4040],b[4040],c[4040],d[4040];

int main() {
    int n;
    scanf("%d",&n);

    for(int i=0; i<size; i++)
        hash[i] = MAX;
    memset(sum,0,sizeof(sum));

    for(int i=0; i<n; i++) {
        scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
    }

    for(int i=0; i<n; i++) {
        for(int j=0; j<n; j++) {
            hash_insert(a[i]+b[j]);
        }
    }

    int ans = 0;
    for(int i=0; i<n; i++) {
        for(int j=0; j<n; j++) {
            ans+=hash_Find(-(c[i]+d[j]));
        }
    }

    cout<<ans<<endl;
    return 0;
}