题目描述

C++代码

样例

#include<iostream>
#include<algorithm>
#include<vector>
#include<cmath>
#include<unordered_map>
using namespace std;
unordered_map<int, bool> mp;
bool prime(int x)
{
    if (x == 1)
    {
        return false;
    }
    for (int i = 2; i <= sqrt(x); i++)
    {
        if (x % i == 0)
        {
            return false;
        }
    }
    return true;
}
int find(int x,int size)
{
    if(!mp[x%size])
    {
        mp[x%size]=true;
        return x%size;
    }
    else{
        int idx=1;
        int start=x%size;
        int k=x%size;
        bool flag=true;
        while(1)
        {
            if(!mp[k])
            {
                mp[k]=true;
                return k;
            }
            if (idx == size)
            {
                flag = false;
                break;
            }
            k=(start+idx*idx)%size;
            idx++;
        }
        return -1;
    }
}
int main()
{
    int m, n;
    cin >> m >> n;
    vector<int> vc(n);
    for (int i = 0; i < n; i++)
    {
        cin >> vc[i];
    }
    while (1)
    {
        if (prime(m))
        {
            break;
        }
        else {
            m++;
        }
    }
    for (int i = 0; i < n; i++)
    {
        if (i < n - 1)
        {
            int f=find(vc[i],m);
            if(f!=-1)
            {
                cout << f << " ";
            }
            else{
                cout << "- ";
            }
        }
        else {
            int f=find(vc[i],m);
            if(f!=-1)
            {
                cout << f;
            }
            else{
                cout << "-";
            }
        }
    }
    return 0;
}