7-2 整型关键字的平方探测法散列

本题的任务很简单：将给定的无重复正整数序列插入一个散列表，输出每个输入的数字在表中的位置。所用的散列函数是 H(key)=key%TSize，其中 TSize 是散列表的表长。要求用平方探测法（只增不减，即H(Key)+i2）解决冲突。

注意散列表的表长最好是个素数。如果输入给定的表长不是素数，你必须将表长重新定义为大于给定表长的最小素数。

输入格式：

首先第一行给出两个正整数 MSize（≤104）和 N（≤MSize），分别对应输入的表长和输入数字的个数。随后第二行给出 N 个不重复的正整数，数字间以空格分隔。

输出格式：

在一行中按照输入的顺序给出每个数字在散列表中的位置（下标从 0 开始）。如果某个数字无法插入，就在其位置上输出 -。输出间以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

4 4

10 6 4 15

输出样例：

0 1 4 -

#include<iostream>
#include<algorithm>
#include<cstring>
#include<unordered_map>

using namespace std;
unordered_map<int, int> map;

bool isprime(int x)
{
    if(x <= 1) return false;
    if(x == 2) return true;
    for(int i = 2; i <= x / i; i++)
        if(x % i == 0) return false;
    return true;
}

int main()
{
    int Msize, n, x;
    cin >> Msize >> n;
    for(int i = Msize; ; i++)
    {
        if(isprime(i)) {
            Msize = i;
            break;
        }
    }
    int h[Msize];
    memset(h, -1, sizeof h);
    for(int i = 0; i < n; i++)
    {
        cin >> x;
        int key = x % Msize;
        bool flag = false;
        for(int j = 0; j < (Msize >> 1); j++)
        {
            int u = (key + j * j) % Msize;
            if(h[u] == -1)
            {
                h[u] = x;
                cout << u;
                if(i != n - 1) cout << ' ';
                flag = true;
                break;
            }
        }
        if(!flag) {
            cout << '-';
            if(i != n - 1) cout << ' ';
        }
    }
    return 0;
}