【HDU 4334 Trouble 】哈希表

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4334

题目大意：输入5行，每行有n个数字，让你从这5行中每行抽取一个数让这5个数的和我0。如果存在输出Yes，否则输出No

 

解题思路：

用O（n^2）的时间求出前两行所有sum的值存入hash表中，注意存的时候存它的相反数，下面再解释。 

然后再用O（n^3）的时间枚举后三行所有的sum值，现在只需要判断hash表中是否有它对应的数，因为上面我们存的是相反数，这就是我们上面为什么存相反数的原因了。

如果找到了，直接跳出循环。

还有一点要注意的是，hash表不能赋初值，hash表最好开到它所有数据十倍左右，藐视我就开了两倍大就wrong answer 好多次。

 

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

const int mod=401704;  //这里开到所有数据十倍左右大
__int64 a[5][210];
__int64 hash[mod];
bool color[mod];

int judge(__int64 x)
{
    int s=x%mod;
    if(s<0) s+=mod;
    while(color[s]&&hash[s]!=x)
        s=(s+1)%mod;
    return s;
}

int main()
{
    int  T, n;
    cin >> T;
    while(T--)
    {
        scanf("%d",&n);
        memset(color,0,sizeof(color));
        for(int i=0; i<5; i++)
            for(int j=0; j<n; j++)
               scanf("%I64d",&a[i][j]);
        for(int i=0; i<n; i++)
            for(int j=0; j<n; j++)
            {
                __int64 sum=-(a[0][i]+a[1][j]);
                int h=judge(sum);
                color[h]=1;
                hash[h]=sum;
            }
        bool ok=false;
        for(int i=0; i<n&&!ok; i++)
           for(int j=0; j<n&&!ok; j++)
              for(int k=0; k<n&&!ok; k++)
              {
                  __int64 sum=a[2][i]+a[3][j]+a[4][k];
                  int h=judge(sum);
                  if(color[h])
                  {
                      ok=true; break;
                  }
              }
        if(ok) printf("Yes\n");
        else printf("No\n");
    }
    return 0;
}