hdu 1880 魔咒词典（双hash）

魔咒词典
Time Limit: 8000/5000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 16775    Accepted Submission(s): 3963

Problem Description
哈利波特在魔法学校的必修课之一就是学习魔咒。据说魔法世界有100000种不同的魔咒，哈利很难全部记住，但是为了对抗强敌，他必须在危急时刻能够调用任何一个需要的魔咒，所以他需要你的帮助。

给你一部魔咒词典。当哈利听到一个魔咒时，你的程序必须告诉他那个魔咒的功能；当哈利需要某个功能但不知道该用什么魔咒时，你的程序要替他找到相应的魔咒。如果他要的魔咒不在词典中，就输出“what?”
 

Input
首先列出词典中不超过100000条不同的魔咒词条，每条格式为：

[魔咒] 对应功能

其中“魔咒”和“对应功能”分别为长度不超过20和80的字符串，字符串中保证不包含字符“[”和“]”，且“]”和后面的字符串之间有且仅有一个空格。词典最后一行以“@END@”结束，这一行不属于词典中的词条。
词典之后的一行包含正整数N（<=1000），随后是N个测试用例。每个测试用例占一行，或者给出“[魔咒]”，或者给出“对应功能”。
 

Output
每个测试用例的输出占一行，输出魔咒对应的功能，或者功能对应的魔咒。如果魔咒不在词典中，就输出“what?”
 

Sample Input
[expelliarmus] the disarming charm
[rictusempra] send a jet of silver light to hit the enemy
[tarantallegra] control the movement of one's legs
[serpensortia] shoot a snake out of the end of one's wand
[lumos] light the wand
[obliviate] the memory charm
[expecto patronum] send a Patronus to the dementors
[accio] the summoning charm
@END@
4
[lumos]
the summoning charm
[arha]
take me to the sky
 

Sample Output
light the wand
accio
what?
what?

 

C/C++:

#include <map>
#include <queue>
#include <cmath>
#include <vector>
#include <string>
#include <cstdio>
#include <cstring>
#include <climits>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f3f
using namespace std;
const int Mod = 1e9 + 7;

char str[300], s1[100], s2[200];
map<pair<int, int>, string> m0;
map<pair<int, int>, string> m1;

int get_hash0(char *s)
{
    int sum = 0;
    for (int i = 0; s[i]; ++ i)
        sum = (sum * 117 % Mod + s[i]) % Mod;
    return sum;
}

int get_hash1(char *s)
{
    int sum = 0;
    for (int i = 0; s[i]; ++ i)
        sum = (sum * 131 % Mod + s[i]) % Mod;
    return sum;
}

int main()
{
    int n, a, b;
    m0.clear();
    m1.clear();
    ios::sync_with_stdio(false);
    while (gets(str), str[0] != '@')
    {
        a = 1, b = 0;
        for (a, b; str[a] != ']'; ++ a, ++ b)
            s1[b] = str[a];
        s1[b] = 0;
        b = 0, a += 2;
        for (a, b; str[a]; ++ a, ++ b)
            s2[b] = str[a];
        s2[b] = 0;
        a = get_hash0(s1), b = get_hash1(s1);
        m0[make_pair(a, b)] = s2;
        a = get_hash0(s2), b = get_hash1(s2);
        m1[make_pair(a, b)] = s1;
    }
    scanf("%d", &n);
    getchar();
    while (n --)
    {
        gets(str);
        if (str[0] == '[')
        {
            str[strlen(str) - 1] = 0;
            a = get_hash0(str + 1), b = get_hash1(str + 1);
            if (m0.find(make_pair(a, b)) != m0.end()) cout <<m0[make_pair(a, b)] <<endl;
            else puts("what?");
        }
        else
        {
            str[strlen(str)] = 0;
            a = get_hash0(str), b = get_hash1(str);
            if (m1.find(make_pair(a, b)) != m1.end()) cout <<m1[make_pair(a, b)] <<endl;
            else puts("what?");
        }
    }
    return 0;
}