[数据结构与算法]PTA 7-43 字符串关键字的散列映射 (25 point(s))

给定一系列由大写英文字母组成的字符串关键字和素数P，用移位法定义的散列函数H(Key)将关键字Key中的最后3个字符映射为整数，每个字符占5位；再用除留余数法将整数映射到长度为P的散列表中。例如将字符串AZDEG插入长度为1009的散列表中，我们首先将26个大写英文字母顺序映射到整数0~25；再通过移位将其映射为3×322+4×32+6=3206；然后根据表长得到3206，即是该字符串的散列映射位置。

发生冲突时请用平方探测法解决。

输入格式:

输入第一行首先给出两个正整数N（≤500）和P（≥2N的最小素数），分别为待插入的关键字总数、以及散列表的长度。第二行给出N个字符串关键字，每个长度不超过8位，其间以空格分隔。

输出格式:

在一行内输出每个字符串关键字在散列表中的位置。数字间以空格分隔，但行末尾不得有多余空格。

输入样例1:

4 11
HELLO ANNK ZOE LOLI

结尾无空行

输出样例1:

3 10 4 0

结尾无空行

输入样例2:

6 11
LLO ANNA NNK ZOJ INNK AAA

输出样例2:

3 0 10 9 6 1

?

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

int n, p;
string key;
bool space = false, vis[1009]{false};
map<string, int> loc;

main() {
    cin >> n >> p;
    while(cin >> key){
        // 颠倒字符串
        reverse(key.begin(), key.end());
        // 是否重复查询
        if(loc.count(key) == true)
            cout << " " << loc[key];
        else{
            int pos = 0, base = 0;
            for(int i = 0; i < key.size() && i < 3; i++)
                pos += (key[i] - 'A') * pow(32, i);
            // 冲突 以当前位置探测 取模防止出界
            int tmp = pos;
            while(vis[pos %= p] == true){
                pos = tmp + pow(base, 2);
                if(vis[pos %= p] == true)
                    pos = tmp - pow(base++, 2);
            }
            vis[pos] = true;
            loc[key] = pos;
            cout << (space++ ? " " : "") << loc[key];
        }
    }
}

参考代码