HDU 4334 Trouble

最新推荐文章于 2018-08-05 19:36:06 发布

第八个猴子

最新推荐文章于 2018-08-05 19:36:06 发布

阅读量67

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Yu2012/archive/2012/08/25/2656764.html

本文介绍了一个使用哈希表解决特定三数组求和为零问题的算法实现。通过对两个数组元素求和并存储到哈希表中，再遍历另外三个数组查找是否存在相反数，以此来判断是否有解。文章提供了完整的C++代码实现。

和POJ1840有点像，但这里只需判断有没有解，一个简单的hash就可以搞定，以后不能滥用vector，自己手写邻接表。

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

typedef __int64 LL;
const int MAXN = 205;
const LL prime = 1000007;
int head[prime + 3], e, next[MAXN * MAXN];
LL val[MAXN * MAXN];

LL s1[MAXN], s2[MAXN], s3[MAXN], s4[MAXN], s5[MAXN];

int hash(LL sum)
{
    int key;
    if(sum > 0)
        key = (int)(sum % prime);
    else
        key = (int)(-sum % prime);
    return key ;
}

void add(int key, LL sum)
{
    val[e] = sum;
    next[e] = head[key];
    head[key] = e ++;
}

int main(){
    int N, n, i, j, k, key;
    LL sum;
    scanf("%d", &N);
    while(N --)
    {
        scanf("%d", &n);
        for(i = 0; i < n; i ++) scanf("%I64d", &s1[i]);
        for(i = 0; i < n; i ++) scanf("%I64d", &s2[i]);
        for(i = 0; i < n; i ++) scanf("%I64d", &s3[i]);
        for(i = 0; i < n; i ++) scanf("%I64d", &s4[i]);
        for(i = 0; i < n; i ++) scanf("%I64d", &s5[i]);
        memset(head, -1, sizeof head);
        e = 0;
        for(i = 0; i < n; i ++)
            for(j = 0; j < n; j ++)
            {
                sum = s1[i] + s2[j];
                key = hash(sum);
                add(key, sum);
            }

        bool flag = false;
        for(i = 0; i < n && !flag; i ++)
            for(j = 0; j < n && !flag; j ++)
                for(k = 0; k < n; k ++)
                {
                    sum = s3[i] + s4[j] + s5[k];
                    int s = hash(-sum);
                    for(int x = head[s]; x != -1; x = next[x])
                    {
                        if(sum + val[x] == 0)
                            flag = true;
                        if(flag)
                            break ;
                    }
                }
        if(flag)
            printf("Yes\n") ;
        else
            printf("No\n") ;
    }
    return 0 ;
}