EOJ 1046 4个数的和为0

poj 2785 http://poj.org/problem?id=2785

eoj 1046 http://acm.cs.ecnu.edu.cn/problem.php?problemid=1046

    思路： 散列（hash） + 拉链法去冲突；

    具体过程：
    
    把num[][1-4]按列划分为num[][1-2] 和 num[][3-4];
    
    对    MAXN + num[i][1] + num[j][2]    散列；O(n^2 * len)

    对    MAXN - num[i][3] - num[j][4]    查找；O(n^2 * len)
    
    须使得被散列值为正，故用一大数 MAXN 保护；len为链表平均长度；

    注意：
    
    1.散列时，sum的地址由 hash(sum) = sum % MOD; 给出

    2.MOD取一个大素数的动机是：使得平均查找效率更优；

    也即散列得更均匀；

    3.解决冲突的链表，这里用数组模拟，貌似比malloc快。

    困惑：
        按题意,节点总数最多为 4000*4000,
        
        而我这里只开了NE个节点，觉得不够， 开大了又MLE。。。

#include <stdio.h>
#include <stdlib.h>
#define MAXN (1<<29)
#define N 4005
#define MOD 3999971
#define NE 3999999
typedef struct
{
    int n;
    int cnt;
    int next;
}E;
int ie;
E edge[NE];    //待分配的节点

int num[N][5];
int hash[MOD+5];

 
void init()
{
    for(int i=0; i<=MOD; i++)
        hash[i] = 0;        
    ie = 1;
}
int search(int n)
{
    int a = n%MOD;
    int p = hash[a];
    while(p)
    {
        if(edge[p].n == n)
            return p;
        p = edge[p].next;
    }
    return 0;    //未找到
}
 
void add(int n)
{
    int a = n%MOD;
    int p;
    if(p = search(n))    //找到，则仅将次数++
        edge[p].cnt ++;
            
    else        //未找到，则申请新节点，并插在头部。
    {
        edge[ie].n = n;
        edge[ie].cnt = 1;
        
        edge[ie].next = hash[a];
        hash[a] = ie++;
    }
 
}
 
int main()
{
    int n;
    while(scanf("%d", &n) != EOF)
    {
        int i, j, cnt = 0;
        for(i=1; i<=n; i++)
            for(j=1; j<=4; j++)
                scanf("%d", &num[i][j]);
        
        init();
        
        for(i=1; i<=n; i++)
            for(j=1; j<=n; j++)        
               add(MAXN + num[i][1] + num[j][2]);
        
        for(i=1; i<=n; i++)
            for(j=1; j<=n; j++)
            {
                int p = search(MAXN - num[i][3] - num[j][4]);
                if(p)
                    cnt += edge[p].cnt;
            }                    
        
        printf("%d\n", cnt);
    }
    
}