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

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

输出样例2:

3 0 10 9 6 1

#define FRER() freopen("in.txt","r",stdin)
#define FREW() freopen("out.txt","w",stdout)
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <map>
#include <stack>
#include <set>
#include <vector>
#include <cstring>
#include <queue>
using namespace std;
const int maxn = 1e5 + 7;
int a[maxn];
int p;
map<string,int>mmp;
int Hash(string s){
	int len = s.size();
	if(len==1) return (s[0]-'A')%p;
	if(len==2) return ((s[0]-'A')*32+(s[1]-'A'))%p;
	return ((s[len-3]-'A')*32*32+(s[len-2]-'A')*32+(s[len-1]-'A'))%p;
}
int main() {
    //FRER();
    //FREW();
	memset(a,0,sizeof a);
	int n;
	scanf("%d%d",&n,&p);
	for(int i=1;i<=n;i++){
		string s;
		cin>>s;
		if(mmp[s]){
		    if(i==1) cout<<mmp[s];
				else cout<<" "<<mmp[s];
				continue;
			}
			int key = Hash(s);
			if(!a[key]){
				mmp[s] = key;
				a[key] = 1;
				if(i==1) cout<<mmp[s];
				else cout<<" "<<mmp[s];
			}else{
				for(int j=0;j<=p/2;j++){
				int k1 = (key + j*j)%p;
				if(!a[k1]){
					mmp[s] = k1;
				    if(i==1) cout<<mmp[s];
					else cout<<" "<<mmp[s];
					a[k1] = 1;
					break;
				}
				int k2 = key - j*j;
				while(k2<0) k2+=p;
				k2%=p;
				if(!a[k2]){
				    mmp[s]= k2;
					if(i==1) cout<<mmp[s];
					else cout<<" "<<mmp[s];
					a[k2] = 1;
					break;
				}
			}
		}
	}
	cout<<endl;
}