7-15 字符串关键字的散列映射 （25 分）(PTA)

7-15 字符串关键字的散列映射 （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

#include <set>
#include <map>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<long, long> pll;
const int inf = 0x3f3f3f3f;
#define mem(a,b) memset(a, b, sizeof(b))
#define FRER freopen("i.txt", "r", stdin)
#define GO int T; scanf("%d", &T); while(T--)
#define IOS ios::sync_with_stdio(false); cin.tie(0)
const int sz = (int)1e6 + 7;
const int maxn = 510;
int v[sz];
string s;
map<string, int> mp;
int n, p;
int idx(char c) {
    return c - 'A';
}
int find(int key) {
    int k = 0, kk = key;
    while(v[kk]) {
        k++;
        int d = k * k;
        kk = (key + d) % p;
        if(!v[kk]) return kk;
        kk = (key - d + p) % p;
    }
    return kk;
}
int main() {
    //IOS;
    //FRER;
    cin >> n >> p;
    mem(v, 0);
    while(n--) {
        cin >> s;
        int len = (int)s.length();
        int cur = min(3, len);
        int key = 0;
        for(int i = len - cur; i < len; i++) {
            key = key*32 + idx(s[i]);
        }
        key = key % p;
        //cout << "(" <<  key << ")" << endl;
        if(mp.count(s)) {
            cout << mp[s];
        }
        else {
            if(v[key]) {
                key = find(key);
//                if(key == -1) return -1;
                v[key] = 1;
                cout << key;
            }
            else {
                v[key] = 1;
                cout << key;
            }
            mp[s] = key;
        }
        cout << (n == 0 ? '\n' : ' ');
    }
    return 0;
}