hdu 4334 Trouble(2012多校联赛4，第3题)

题目：hdu 4334 Trouble 这个是2012 hdu 多校联赛的一个题目。

题意：给五组数，让你判断是否存在：从每组中选择一个，然后这五个树和是0.也就是s[0][i]+s[1][j]+s[2][k]+s[3][l]+s[4][f]=0；

解题：由于数据比较大，如果用普通思路直接5层循环的话，可能是要超时的，这就用到了二分的思想。题目要求是s[0][i]+s[1][j]+s[2][k]+s[3][l]+s[4][f]=0，我们可以变成是s[0][i]+s[1][j]+s[2][k]=-(s[3][l]+s[4][f])；这样时间复杂都就会变小近一半了。当时在比赛的时候，我看到这个题目就想到这个方法，但是实现的时候没有想到用hash table（哈希表），导致开始是一直MLE，后来是一直TLE。比赛下来和队里面的讨论了一下，用hash table就AC了。

代码：

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <vector>

using namespace std;
#define N 40000
#define M 40000
//#define LL long long
#define LL __int64
vector <LL> hash[N];
LL s[5][205];
int num;
void init()
{
    int i,j;
    scanf("%d",&num);
    for(i=0;i<N;i++)
        hash[i].clear();
    for(i=0;i<5;i++)
        for(j=0;j<num;j++)
            scanf("%I64d",&s[i][j]);
}
bool Find(LL key)
{
    int i,length;
    LL k=key;
    if(key<0)
    k=-key;
    k=k%M; length=hash[k].size();
    for(i=0;i<length;i++)
    {
        if(key==hash[k][i])
        return true;
    }
    return false;
}
void Add(LL key)
{
    //if(Find(key))    return ;//加点的时候判断是已经存在，也就是判断是否重复。加上不来想优化的，结果时间更长了
    LL k=key;
    if(key<0)
    k=-key;
    hash[k%M].push_back(key);
}

bool fun()
{
    int i,j,k;
    for(i=0;i<num;i++)
        for(j=0;j<num;j++)
            Add(s[0][i]+s[1][j]);
            
    for(i=0;i<num;i++)
        for(j=0;j<num;j++)
            for(k=0;k<num;k++)
                if(Find(-(s[2][i]+s[3][j]+s[4][k])))
                return true;
     
    return false;
}

int main()
{
    //freopen("/home/acm/JPY/input.txt","r",stdin);
    int T;    cin>>T;
    while(T--)
    {
        init();
        if(fun())        printf("Yes\n");
        else        printf("No\n");
    }
}