PAT 1116. Come on! Let's C

PAT 1116. Come on! Let’s C (20)
https://www.patest.cn/contests/pat-a-practise/1116

#include <bits/stdc++.h>
using namespace std;
bool isPrime(int x)
{
    if (x < 2) return false;
    for (int i = 2; i * i <= x; i++) {
        if (x % i == 0) return false;
    }
    return true;

}
int main()
{
    int N;
    scanf("%d\n", &N);
    map<string, int> mp;
    for (int i = 1; i <= N; i++) {
        string s;
        cin >> s;
        mp[s] = i;
    }
    int K;
    scanf("%d\n", &K);
    while (K--) {
        string s;
        cin >> s;
        cout << s << ": ";
        if (mp.count(s)) {
            if (mp[s] == 0) puts("Checked"); else
            if (mp[s] == 1) puts("Mystery Award"); else
            if (isPrime(mp[s])) puts("Minion"); else
            puts("Chocolate");
            mp[s] = 0;
        } else {
            puts("Are you kidding?");
        }
    }
}