SP1296 SUMFOUR - 4 values whose sum is 0

题目链接

4 values whose sum is 0

题意：

从四个序列中取一个数，问其相加有多少种可能。
$$a_i+b_i+c_i+d_i=0$$
即求
$$a_i+b_i=-c_i-d_i$$
至此，我们可以选择ai+bi或-ci-di存入哈希表，并用另外一个值去到哈希表中寻找。只不过本题可以用邻接表来优化哈希表（至少我是这么做的），注意因为答案可能会有负数，所以取模需要加上mod值再进行取模。

一些注释看代码吧。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;
int n,a[4001],b[4001],c[4001],d[4001];
const int mod=7999997;
int ans=0,sum;
struct node{
    int u,v;
}e[4001*4001];
int head[mod],cnt=0;
void add(int a,int b){
    //cout<<a<<" "<<b<<endl;
    e[++cnt].u=head[a];
    head[a]=cnt;
    e[cnt].v=b;
}//像极了链式存图
void find(int x){
    int u=(x%mod+mod)%mod;
    for(int i=head[u];i;i=e[i].u){
        int v=e[i].v;
        if(v==x)ans++;
    }
}//和访问图的节点一样去访问哈希表内的元素
int main(){
    scanf("%d",&n);
    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++){
            sum=a[i]+b[j];
            //cout<<sum<<endl;
            add((sum%mod+mod)%mod,sum);//存入哈希表，注意是先存哈希值再存键值
        }
    }
    //cout<<"done"<<endl;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            int need=-c[i]-d[j];
            find(need);//访问哈希表内的元素
            //cout<<ans<<endl;
        }
    }
    //cout<<ans<<endl;
    printf("%d\n",ans);
    return 0;
}