1116 Come on! Let's C (20分)

四种情况：第一、素数、没有、出现过

思考下如何分辨四种情况即可。

 

#include<bits/stdc++.h>
using namespace std;
map<int,int> r;
bool isPrime(int n){
    for(int i=2;i<=sqrt(n);i++){
        if(n%i==0) return false;
    }
    return true;
}
int main(){
    //4种情况
    int n,k;
    cin>>n;
    for(int i=1;i<=n;i++){
        int x;cin>>x;
        r[x]=i;
    }
    cin>>k;
    while(k--){
        int x;cin>>x;
        if(r.find(x)==r.end()){
            printf("%04d: Are you kidding?\n",x);
        }else{
            if(r[x]==-1){
                printf("%04d: Checked\n",x);
            }else if(r[x]==1){
                printf("%04d: Mystery Award\n",x);r[x]=-1;
            }else if(isPrime(r[x])){
                printf("%04d: Minion\n",x);r[x]=-1;
            }else{
                printf("%04d: Chocolate\n",x);r[x]=-1;
            }
        }
    }
    return 0;
}