PTA 7-42 整型关键字的散列映射 (25 point(s))

 给定一系列整型关键字和素数P，用除留余数法定义的散列函数H(Key)=Key将关键字映射到长度为P的散列表中。用线性探测法解决冲突。

输入格式:

输入第一行首先给出两个正整数N（≤1000）和P（≥N的最小素数），分别为待插入的关键字总数、以及散列表的长度。第二行给出N个整型关键字。数字间以空格分隔。

输出格式:

在一行内输出每个整型关键字在散列表中的位置。数字间以空格分隔，但行末尾不得有多余空格。

输入样例:

4 5
24 15 61 88

结尾无空行

输出样例:

4 0 1 3

结尾无空行

#include<bits/stdc++.h>
using namespace std;

main() {
	map<int, int> loc;
	int N, P, key, space = false, vis[1009]{false};
	cin >> N >> P;
    
	while (cin >> key) {
		if (loc.count(key) == true)
			cout << " " << loc[key];
		else {
			int pos = key;
			while (vis[pos %= P] == true) pos++;

            vis[pos] = true;
			loc[key] = pos % P;
			cout << (space++ ? " " : "") << loc[key];
		}
	}
}

参考代码