[Aizu] ITP2_8_A: Map: Search

题目

传送门: ITP2_8_A: Map: Search

描述

对于存储由一对字符串键和整数值形成的元素的字典M, 执行下面给出的操作, 请注意, M中的每一个键必须是唯一的

• \(insert(key, x)\): 插入一个由键和值组成的元素到M中, 如果已经有元素包含了这个键, 替换那个元素的值为x
• \(get(key)\): 输出一个指定的键对应的值

输入

输入以下列格式给出

\(q\)
\(query_{1}\)
\(query_{2}\)
\(.\)
\(.\)
\(query_{n}\)

每一个\(query_{i}\)都为:
\(0\) \(key\) \(x\)
或
\(1\) \(key\)
其中第一个数字0和1分别表示 insert 和 get 操作.

输出

对于每一个 get 操作, 在一行输出一个整数

限制条件

• \(1 \leq q \leq 200000\)
• \(1 \leq x \leq 1000000000\)
• \(1 \leq\) length of \(key \leq 20\)
• \(key\)全部由小写字母组成
• 对于每一个 get 操作, 包含指定键值的元素一定存在于M中.

样例输入

7
0 blue 4
0 red 1
0 white 5
1 red
1 blue
0 black 8
1 black

样例输出

1
4
8

求解

自己实现

分析

关于哈希表的经典练习题, 不做多的说明

设计

由于请求的数量不超过200000, 为了使哈希表的效率更高, 空间分配为它的二倍左右
创建一个结构体 Ele 用于存储键值对构成的元素
设计一个hash函数, 计算出字符串对应的值
为了简单, 直接写了insert和get两个普通的函数, 直接操作全局变量的数组, 内部的移动也是随便给了数字, 担心所有的都聚在前面, 后面空出来, 万幸通过了

编码

#include <iostream>
#include <string>
using namespace std;
#define MAX_Q 400005

struct Ele {
    string key;
    int value;
    Ele() {
        key = "";
        value = 0;
    }
    Ele(string str, int i) {
        key = str;
        value = i;
    }
};

Ele space[MAX_Q];

long get_key(string str) {
    long total = 0;
    for (int i = 0; i < str.length(); i++) {
        total = (total * 4 + str[i] - 'a' + 1) % 1000000007;
    }
    return total;
}

void insert(string key, int value) {
    int index = get_key(key) % MAX_Q;
    while (space[index].key != "" && space[index].key != key) {
        index = (index + 17541) % MAX_Q;
    }
    Ele m(key, value);
    space[index] = m;
}

int search(string key) {
    int index = get_key(key) % MAX_Q;
    while (space[index].key != "" && space[index].key != key) {
        index = (index + 17541) % MAX_Q;
    }
    if (space[index].key == key) {
        return space[index].value;
    } else {
        return 0;
    }
} 

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(0);

    int q;
    cin >> q;
    int com, value;
    string key;
    while (q--) {
        cin >> com >> key;
        if (com == 0) {
            cin >> value;
            insert(key, value);
        } else {
            cout << search(key) << endl;
        }
    }
}

结果

总结

之前错了4次
第一次: 审题不清, 没有看清楚说键对应的值需要更新, 就直接往里扔, 导致get到了错误的元素
第二次: 指针只往后走, 忘了需要模数组大小, 然后出错
第三次: 对于一些长的字符串, 超出了计算出来的hash值超出了int的表示范围
第四次: 虽然修改了计算出的hash值, 使其输出为long, 但是里面的临时值却设置成了int, 导致临时值在还没来得及求模之前就已经溢出

STL

编码

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

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(0);

    int q, com, x;
    string key;
    map<string, int> my_map;

    cin >> q;
    while (q--) {
        cin >> com >> key;
        if (com == 0) {
            cin >> x;
            my_map[key] = x;
        } else {
            cout << my_map[key] << endl;
        }
    }
}

结果

备注

哎, 我自己的居然时间和空间都能比模版好一丢丢, 莫名开心