Uva-1152 (Hash) 4 Values whose Sum is 0

13 篇文章 0 订阅

题目链接:https://cn.vjudge.net/problem/UVA-1152

题目解法有多种,这里只讨论使用hash表实现,下面给出两种使用hash表解决此题的代码:
Hash写法一:
AC:Time 890ms
先枚举A集合和B集合中元素的和,直接存入Hash表中,接下来枚举C集合和D集合中元素的和的相反数在Hash表中出现的次数,即可,具体在代码中解释。

#include <bits/stdc++.h>
using namespace std;

struct Hash{
    static const int hashsize = 0x7fffff;
    //data中存放着所有集合A和集合B中元素的和的值,num中就是存放该值的出现的次数
    int data[hashsize+1], num[hashsize+1];
    void clear(){ memset(num, 0, sizeof(num)); }

    int& operator[] (int k){
        int i;
        //i = k & hashsize 实际上就是取k的二进制的后23位,作为该值的hash值,
        //由于可能存在多个值会共享同一个hash值,若某次查询我们发现该hash值对应
        //的num非零,就说明该hash值已经被占用,所以我们就可以从此处往后遍历,直
        //到找到某个地方是未被使用过,即num值为零,或在途中找到了相同的data值,
        //就说明之前这个k值出现过,返回即可。
        for(i = k&hashsize; num[i] && data[i] != k; i = (i+1)&hashsize);
        data[i] = k;
        return num[i];
    }
};

const int maxn = 4500;
int a[4][maxn];
int ab[maxn*maxn];
Hash Map;

int main(){
    int T;
    scanf("%d", &T);
    while(T--){
        int n;
        scanf("%d", &n);

        Map.clear();

        for(int i = 0; i < n; ++i)
        for(int j = 0; j < 4; ++j)
            scanf("%d", &a[j][i]);

        for(int i = 0; i < n; ++i)
        for(int j = 0; j < n; ++j)
            Map[a[0][i]+a[1][j]]++;

        int ans = 0;
        for(int i = 0; i < n; ++i)
        for(int j = 0; j < n; ++j)
            ans += Map[-a[2][i]-a[3][j]];

        printf("%d\n", ans);

        if(T) printf("\n");
    }
    return 0;
}

Hash写法二:
AC:Time 2800ms
先将枚举A集合和B集合中元素的和,存入数组ab中,设置hashsize大小为10000007,我们将ab数组中的每个元素都映射到hash表中去,接下来我们枚举C集合和D集合中的元素的和的相反数,命名为target,在hash表中寻找有多少个target,即可,这其中有细节,就是我们默认绝对值相同的元素,hash值也相同,这样就可以很方便的处理负数了。
代码如下:

#include <bits/stdc++.h>
using namespace std;

const int maxn = 5000;
const int hashsize = 10000007;
int n, a[4][maxn];
int ab[maxn*maxn];
int head[hashsize], Next[maxn*maxn];

void init_head(){
    memset(head, -1, sizeof(head));
}

void to_insert(int s){
    int h = abs(ab[s])%hashsize;
    Next[s] = head[h];
    head[h] = s;
}

int to_find(int target){
    int h = abs(target)%hashsize;
    int u = head[h];
    int num = 0;
    while(u != -1){
        if(ab[u] == target) num++;
        u = Next[u];
    }
    return num;
}

int main(){
    int T;
    scanf("%d", &T);
    for(int ca = 0; ca < T; ++ca){
        if(ca) puts("");
        scanf("%d", &n);
        for(int j = 0; j < n; ++j)
            for(int i = 0; i < 4; ++i)
                scanf("%d", &a[i][j]);

        init_head();
        for(int i = 0; i < n; ++i)
            for(int j = 0; j < n; ++j){
                ab[i*n + j] = a[0][i] + a[1][j];
                to_insert(i*n + j);
            }

        int ans = 0;
        for(int i = 0; i < n; ++i)
            for(int j = 0; j < n; ++j)
                ans = ans + to_find(-a[2][i]-a[3][j]);

        printf("%d\n", ans);
    }
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值