PAT甲級 1078

題目 Hashing

The task of this problem is simple: insert a sequence of distinct positive integers into a hash table, and output the positions of the input numbers. The hash function is defined to be H(key)=key%TSize where TSize is the maximum size of the hash table. Quadratic probing (with positive increments only) is used to solve the collisions.
Note that the table size is better to be prime. If the maximum size given by the user is not prime, you must re-define the table size to be the smallest prime number which is larger than the size given by the user.
Input Specification:
Each input file contains one test case. For each case, the first line contains two positive numbers: MSize (≤10^​4​​ ) and N (≤MSize) which are the user-defined table size and the number of input numbers, respectively. Then N distinct positive integers are given in the next line. All the numbers in a line are separated by a space.
Output Specification:
For each test case, print the corresponding positions (index starts from 0) of the input numbers in one line. All the numbers in a line are separated by a space, and there must be no extra space at the end of the line. In case it is impossible to insert the number, print “-” instead.

解析

輸入msize和n，如果msize不爲素數，找到比msize的最小素數。最小素數可能會大於給定的10^4

代碼

#include<bits/stdc++.h>

#define INF 1<<29

using namespace std;

const int maxn = 100005;
int prime[maxn], pNum = 0;

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

void pat1078() {
    int msize, n;
    int flag[maxn];
    memset(flag, 0, sizeof(maxn));
    cin >> msize >> n;
    if (!is_prime(msize)) {
        for (int i = msize + 1; i < maxn; ++i) {
            if (is_prime(i)) {
                msize = i;
                break;
            }
        }
    }
    for (int i = 0; i < n; ++i) {
        int num;
        scanf("%d", &num);
        int temp = -1;
        bool f = false;
        for (int j = 0; j <= msize; ++j) {
            temp = (num + j * j) % msize;
            if (flag[temp] == 0) {
                flag[temp] = 1;
                f = true;
                break;
            }
        }
        if (f) {
            printf("%d", temp);
        } else {
            printf("-");
        }

        if (i != n - 1) {
            printf(" ");
        }
    }

}


int main() {
    pat1078();
    return 0;
}