PAT 1059 C语言竞赛（20 分）

1059 C语言竞赛（20 分）

C 语言竞赛是浙江大学计算机学院主持的一个欢乐的竞赛。既然竞赛主旨是为了好玩，颁奖规则也就制定得很滑稽：
0、冠军将赢得一份“神秘大奖”（比如很巨大的一本学生研究论文集……）。
1、排名为素数的学生将赢得最好的奖品 —— 小黄人玩偶！
2、其他人将得到巧克力。
给定比赛的最终排名以及一系列参赛者的 ID，你要给出这些参赛者应该获得的奖品。
输入格式：
输入第一行给出一个正整数 N（$≤10^4$​ ），是参赛者人数。随后 N 行给出最终排名，每行按排名顺序给出一位参赛者的 ID（4 位数字组成）。接下来给出一个正整数 K 以及 K 个需要查询的 ID。
输出格式：
对每个要查询的 ID，在一行中输出 ID: 奖品，其中奖品或者是 Mystery Award（神秘大奖）、或者是 Minion（小黄人）、或者是 Chocolate（巧克力）。如果所查 ID 根本不在排名里，打印 Are you kidding?（耍我呢？）。如果该 ID 已经查过了（即奖品已经领过了），打印 ID: Checked（不能多吃多占）。
输入样例：

6
1111
6666
8888
1234
5555
0001
6
8888
0001
1111
2222
8888
2222

输出样例：

8888: Minion
0001: Chocolate
1111: Mystery Award
2222: Are you kidding?
8888: Checked
2222: Are you kidding?

解析

这题时间挺紧张的。本来我是先生成一个素数表的，但是运行超时了。
本来我是用string存储ID的，这样输出的时候很轻松，但是还是超时了。
所以程序流程是；
①处理排名输入。放在vector里。 还要添加bool checked。存储是否被查询过。
②处理查询输入。放在数组里。
查询过Checked
如果是第一名：Mystery Award
如果是素数名称Minion
不是素数名次 Chocolate
不存在Are you kidding?
每查询过一次，就把bool checked设置为true。代表已被查询过

code:

#include<bits/stdc++.h>
using namespace std;
struct rating{
    int id;
    bool checked;
};
bool prime(int N){
    for(int i=2;i<=sqrt(N);i++)
        if(N%i==0)
            return false;
    return true;
}
int main()
{
    int N;
    cin>>N;
    vector<rating> data;
    rating temp;
    for(int i=0;i<N;i++){
        cin>>temp.id;
        temp.checked = false;
        data.push_back(temp);
    }
    int K;
    cin>>K;
    int find_date[K];
    for(int i=0;i<K;i++)
        cin>>find_date[i];
    for(int i=0;i<K;i++){
        int j;
        for(j=0;j<N;j++){
            if(find_date[i]==data[j].id){
                if(data[j].checked == true)
                    printf("%04d: Checked\n",data[j].id);
                else{
                    if(j==0)
                        printf("%04d: Mystery Award\n",data[j].id);
                    else if(prime(j+1) == true)
                        printf("%04d: Minion\n",data[j].id);
                    else
                        printf("%04d: Chocolate\n",data[j].id);
                    data[j].checked=true;
                }
                break;
            }
        }
        if(j==N)
            printf("%04d: Are you kidding?\n",find_date[i]);
    }
}