7-43 字符串关键字的散列映射 (25 分)

给定一系列由大写英文字母组成的字符串关键字和素数P，用移位法定义的散列函数H(Key)将关键字Key中的最后3个字符映射为整数，每个字符占5位；再用除留余数法将整数映射到长度为P的散列表中。例如将字符串AZDEG插入长度为1009的散列表中，我们首先将26个大写英文字母顺序映射到整数0~25；再通过移位将其映射为3×32​2​​+4×32+6=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<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<set>
#include<map>
using namespace std;
#define maxn 1000007
#define INF 0xffffff
#define mod = 1e6+7;

map<string, int> mp;
int vis[maxn] = {0};

int main() {
    string s;
    int n, m;
    cin>>n>>m;
    for(int i = 1; i <= n; ++i) {
        cin >> s; 
        int len = s.size();
        int num = 0;
        if(len == 1) {
            num += (s[0]-'A');
        }
        else if(len == 2) {
            num = 32*(s[0]-'A') + (s[1]-'A');
        }
        else {
            for(int j = 3; j >= 1; --j) {
                int pos = len - j;
                num = num * 32 + s[pos] - 'A';
            }
        }
        num %= m;
        if(mp[s] == 0 && vis[num]) {
            for(int t = 1; t < maxn; ++t) {           //平方探测处理冲突
                if(!vis[(num+t*t)%m]) {
                    num = (num+t*t)%m;
                    vis[num] = true;
                    mp[s] = num;
                    cout << num;
                    break;
                }
                else if(!vis[(num-t*t+m)%m]) {
                    num = (num-t*t+m)%m;
                    vis[num] = true;
                    mp[s] = num;
                    cout << num;
                    break;
                }
            }
        }
        else {
            num %= m;
            mp[s] = num;
            cout << num;
            vis[num] = true;
        }
        if(i < n) 
            cout << " ";
        else 
            cout << endl;
    }
    return 0;
}