PAT (Advanced) 1078. Hashing (25)

注意是平方探测，1^2,2^2,3^2.....其中要平方的数小于TSize，不然探测没法停止，我之前在这里卡着过不了。

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <algorithm>
#include <cmath>
#include <string>
#include <cstring>

using namespace std;

bool is_prime(int n)
{
	if (n == 0 || n == 1)
		return false;
	if (n == 2)
		return true;
	int bound = (int)sqrt(n);
	for (int i = 2; i <= bound; i++)
	{
		if (n % i == 0)
			return false;
	}
	return true;
}

int get_TSize(int MSize)
{
	for (int i = MSize;; i++)
	{
		if (is_prime(i))
			return i;
	}
}
int main()
{
	freopen("in.txt", "r", stdin);
	int MSize, N;
	cin >> MSize >> N;
	int TSize = get_TSize(MSize);
	int *flag = new int[TSize];
//	bool *flag = new bool[TSize];
	memset(flag, 0, sizeof(flag) * TSize);

	for (int i = 0; i < N; i++)
	{
		int tmp;
		cin >> tmp;
		int pos = tmp % TSize;
		if (i == 0)
		{
			flag[pos] = true;
			cout << pos;
		}
		else
		{
			if (flag[pos])
			{
				bool inst = false;
				for (int ins = 1;ins < TSize; ins++)
				{
					if (!flag[(pos + ins*ins) % TSize])
					{
						inst = true;
						flag[(pos + ins*ins) % TSize] = true;
						cout << " " << (pos + ins * ins) % TSize;
						break;
					}
				}
				if (!inst)
					cout << " -";
			}
			else
			{
				flag[pos] = true;
				cout << " " << pos;
			}
		}
	}
	cout << endl;
}