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

如有不对，不吝赐教
进入正题：
给定一系列由大写英文字母组成的字符串关键字和素数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冲突，Hash函数已经由题目给出了，所以只要注意一下平方探测法是 正负1，正负4，正负9……来进行探测的就好。

下面给出代码：

#include<stdio.h>
#include<string.h>

#define arg1 (32*32)
#define arg2 (32)       //移位法的两个参数

int Insert(char str[][9],char *input,int P);

int main(void)
{
    int N,P;
    int i;
    scanf("%d %d",&N,&P);
    char str[P][9];
    char input[9];
    int position[N];     //记录位置

    for(i=0;i<P;i++)
      str[i][0]='\0';   //初始化Hash表

    for(i=0;i<N;i++){
      scanf("%s",input);
      position[i]=Insert(str,input,P);
    }

    printf("%d",position[0]);
    for(i=1;i<N;i++)
      printf(" %d",position[i]);

    return 0;
}

int Insert(char str[][9],char *input,int P)
{
    int i=0;
    int flag=1;
    int key,pos;

    while(strlen(input)>3&&input[i+3]!='\0')
      i++;                  //找到识别位点

    if(1==strlen(input))
      key=(input[0]-'A');
    else if(2==strlen(input))
      key=(input[0]-'A')*arg2+(input[1]-'A');
    else
      key=(input[i]-'A')*arg1+(input[i+1]-'A')*arg2+(input[i+2]-'A');

    key%=P;  //按照移位法得到键值

    i=1;
    pos=key;
    while('\0'!=str[pos][0]&&strcmp(input,str[pos])){
      pos=key+flag*i*i;        //该位置已有并且不为input
      flag*=-1;             //变成反的
      if(flag==1)
        i++;                //当flag变成正数的时候 i得++ 探测下一个
      if(pos>=P)
        pos%=P;             //大于P
      if(pos<0)
        pos+=P;             //小于0
    }
    strcpy(str[pos],input);   //填入Hash表

    return pos;
}

测试结果：