PTA 字符串关键字的散列映射

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

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

输入格式:

输入第一行首先给出两个正整数$N$（$\le 500$）和$P$（$\ge 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

解题思路：
（1） 字符串长度有可能小于3 （2） 字符串有可能有重复的，遇到重复的直接输出




#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<map>
using namespace std;
int n,p,vis[2020];
map<string,int> mp;
int main(){
	int i,j;
	memset(vis,0,sizeof(vis));
	char ch[20];
	scanf("%d%d",&n,&p);
	for(i=0;i<n;i++){
		scanf("%s",ch);
		if(mp[ch]){
			printf("%d",mp[ch]-1);
			if(i!=n-1)
				printf(" ");
			continue;
		}
		int len=strlen(ch);
		int t1=0,t2=0,t3=0;
		if(len==1){
			t1=ch[len-1]-'A';
		}
		if(len==2){
			t1=ch[len-1]-'A';t2=ch[len-2]-'A';
		} 
		else if(len>=3){
			t1=ch[len-1]-'A';t2=ch[len-2]-'A',t3=ch[len-3]-'A';
		}
		int sum=t3*32*32+t2*32+t1;
		sum%=p;
		//printf("%d\n",sum);
		int flag=1;
		while(vis[sum]){
			int y=(sum+flag*flag)%p;
			if(!vis[y]){
				mp[ch]=y+1;
				printf("%d",y);
				
				vis[y]=1;
				break; 
			}
			int k=sum-flag*flag;
			k%=p;
			k+=p;
			k%=p;
			if(!vis[k]){
				vis[k]=1;
				mp[ch]=k+1;
				printf("%d",k);
				break;
			}
			flag++;
		}
		if(!vis[sum]){
			vis[sum]=1;
			mp[ch]=sum+1;
			printf("%d",sum);
		}
		if(i!=n-1)
			printf(" ");
		else
			printf("\n");
	}
	return 0;
} 

为什么我总是没想到坑点，这么坑。