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

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

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

输入格式:

输入第一行首先给出两个正整数N（≤）和P（≥的最小素数），分别为待插入的关键字总数、以及散列表的长度。第二行给出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 <cstdio>
#include <map>
#include <cstring>

using namespace std;

int n,p,vis[2001];
int h[4000],ans[500];
char s[10];
map<string,int> mp;
int main() {
    scanf("%d%d",&n,&p);
    for(int i = 1;i <= 1000;i ++) {
        h[i * 2 - 1] = i * i % p;
        h[i * 2] = (-i * i + p * p) % p;
    }
    for(int i = 0;i < n;i ++) {
        scanf("%s",s);
        if(mp.count(s)) {
            ans[i] = mp[s];
            continue;
        }
        int len = strlen(s),ii = 0;
        int d = ((len >= 3 ? s[len - 3] - 'A' : 0) * 32 * 32 + (len >= 2 ? s[len - 2] - 'A' : 0) * 32 + (len >= 1 ? s[len - 1] - 'A' : 0)) % p;
        while(vis[(d + h[ii]) % p]) ii ++;
        vis[(d + h[ii]) % p] = 1;
        ans[i] = (d + h[ii]) % p;
        mp[s] = ans[i];
    }
    for(int i = 0;i < n;i ++) {
        if(i) putchar(' ');
        printf("%d",ans[i]);
    }
}