最小的数

最小的数

Time Limit : 3000/1000ms (Java/Other)   Memory Limit : 65535/32768K (Java/Other)

Total Submission(s) : 29   Accepted Submission(s) : 16

Font: Times New Roman | Verdana | Georgia

Font Size: ← →

Problem Description

Input

本题只有唯一一组测试数据，第一行给出N，q，表示K的个数以及q次询问。1<=N<=1000，q<=10^5.
接下来q行每行一个正整数（64位整数范围内），请判断其对10^9+7取余后，是否在集合K中。

Output

对于每次询问，根据题意输出Yes或No

Sample Input

3 3
2
6
33

Sample Output

Yes
Yes
No


这题并不难，可是一开始的数组被我开太大了，就没反应过来了，用一个 set就过了。

#include <iostream>  
#include <cstdio>
#include <cstring>
#include <cmath>
#include <set>
#include <algorithm>
#define  ll long long int
#define Ma 1000000007
#define N 10005
using namespace std;
ll prime[N],p[N];
int n,m,k=0;
set<ll> s;
void IsPrime() {
    memset(prime, 0, sizeof(prime));
    prime[0] = prime[1] = 1;
    for (int i = 2; i < N; i++){
        if (!prime[i]) {
            p[++k] = i;
            for (int j = 2; j * i < N; j++)
                prime[i * j] = 1;
        }
    }
}

int main(){

    scanf("%d%d",&n,&m);
    IsPrime();
    for(int i=1;i<=n;i++){
        ll ans=1;
        for(int j=1;j<=i;j++){
            ans=p[j]*ans%Ma;
        }
        s.insert(ans);
    }
    while(m--){
     ll h;
        scanf("%lld",&h);

        if(s.count(h)){
            printf("Yes
");
        }else{
            printf("No
");
        }
    }
    return 0;
}