[蓝桥杯]质数的后代

[蓝桥杯][算法提高][VIP]质数的后代

题目描述
在上一季里，曾提到过质数的孤独，其实从另一个角度看，无情隔膜它们的合数全是质数的后代，因为合数可以由质数相乘结合而得。
如果一个合数由两个质数相乘而得，那么我们就叫它是质数们的直接后代。现在，给你一系列自然数，判断它们是否是质数的直接后代。
数据规模和约定
1< =T< =20
2< =要判断的自然数< =10^5
输入
第一行一个正整数T，表示需要判断的自然数数量
接下来T行，每行一个要判断的自然数
输出
共T行，依次对于输入中给出的自然数，判断是否为质数的直接后代，是则输出Yes，否则输出No
样例输入
4
3
4
6
12
样例输出
No
Yes
Yes
No

---

这题目用到了素数打表，我们用p数组记录所有的素数，用a记录每一个数是不是素数，0表示是素数，1表示不是素数。

vector<int >p;
int a[100001] = {0}; //0标志为素数
void getprime() {
    a[1] = a[0] = 1;
    for (int i = 2; i  <= 100000; i++) {
        if (a[i] == 0) {
            p.push_back(i);
            for (int j = i + i; j <= 100000; j += i) {
                a[j] = 1;
            }
        }
    }
}

ps:0,1不是素数。

---

代码如下：

#include<iostream>
#include<vector>
using namespace std;
vector<int >p;
int a[100001] = {0}; //0标志为素数
void getprime() {
    a[1] = a[0] = 1;
    for (int i = 2; i * i <= 100000; i++) {
        if (a[i] == 0) {
            p.push_back(i);
            for (int j = i * i; j <= 100000; j += i) {
                a[j] = 1;
            }
        }
    }
}
void check(int n) {
    for (int i = 0; i <= p.size() - 1; i++) {
        if(n % p[i] == 0 && a[n / p[i]] == 0) {
            cout << "Yes\n";
            return;
        }
    }
    cout << "No\n";
    return;
}
int main () {
    int n, key;
    cin >> n;
    getprime();
    while(n--) {
        cin >> key;
        check(key);
    }
    return 0;
}