PTA 数据结构与算法分析 7-43 字符串关键字的散列映射 （25 分)

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

思路： 线性探测表的定址方式为( Hash(X) + F(i) ) mod TableSize ) ,其中F(i) = i ， 而平方探测法的F(i) = i * i , 知道这个可以开始写代码了

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
int N,P,address[1010],flag = 0;
char str[2010][10];//数组的容量要足够大，不然最后一个测试点过不了

int Hash(int x)
{
    return x%P;
}

int change(char T[]) //将字符串的后三位转化为数字
{
    int i;
    if(strlen(T) < 3){
      i = 0;
    }
    else{
      i = strlen(T) - 3;
    }
    int sum = 0;
    for(;i < strlen(T);i++){
        sum = (sum<<5) + T[i] - 'A';
    }
    return sum;
}

int main()
{
    cin >> N >> P;
    getchar();
    for(int i=0;i < N;i++){
        char s[10];
        scanf("%s",s);
        int n = change(s);
        bool f = false;
        for(int j=0;j < P;j++){
            if(strcmp(str[j],s) == 0){//判断字符串是否在表中
                printf(" %d",j);
                f = true;
                break;
            }
        }
        if(f)
          continue;
        int c = 1,j = 0,tag = 0;
        while(c){
            int vl = n;
            if(!tag){
                vl = (vl + j*j)%P;
                tag = 1;
            }
            else{
                vl = vl - j*j;
                while(vl < 0){ //注意v1小于零的情况，忽略的话3，4两个测试点会错误
                  vl += P;
                }
                vl = vl%P;
                ++j;
                tag = 0;
            }
            if(!address[vl]){
                address[vl] = 1;
                strcpy(str[vl],s);
                printf("%s%d",flag++?" ":"",vl);
                c = 0;
                break;
            }
        }
    }
    return 0;
}