蓝桥杯 ADV-154 算法提高 质数的后代

算法提高 质数的后代
时间限制：1.0s 内存限制：256.0MB

 

问题描述
　　在上一季里，曾提到过质数的孤独，其实从另一个角度看，无情隔膜它们的合数全是质数的后代，因为合数可以由质数相乘结合而得。
　　如果一个合数由两个质数相乘而得，那么我们就叫它是质数们的直接后代。现在，给你一系列自然数，判断它们是否是质数的直接后代。

 

输入格式
　　第一行一个正整数T，表示需要判断的自然数数量
　　接下来T行，每行一个要判断的自然数

 

输出格式
　　共T行，依次对于输入中给出的自然数，判断是否为质数的直接后代，是则输出Yes，否则输出No

 

样例输入
4
3
4
6
12

 

样例输出
No
Yes
Yes
No

 

数据规模和约定
　　1<=T<=20
　　2<=要判断的自然数<=10^5
 

#include <cstdio>
#include <vector>
#include <cstdlib>

using namespace std;

int main()
{
    int T;
    int numbers[22] = { 0 };
    scanf("%d", &T);
    for (int i = 0; i < T; ++i)
        scanf("%d", &numbers[i]);

    bool is_not_prime[100005] = { 0 };
    vector<int> primes({2});
    for (int i = 4; i <= 100000; i += 2)
        is_not_prime[i] = 1;
    for (int i = 3; i <= 100000; i += 2)
    {
        if (is_not_prime[i] == 0)
        {
            primes.push_back(i);
            for (int j = 2 * i; j <= 100000; j += i)
                is_not_prime[j] = 1;
        }
    }

    bool son_of_prime[100005] = { 0 };
    for (int i = 0; i < primes.size(); ++i)
    {
        for (int j = 0; j < primes.size() && primes[i] * primes[j] <= 100000; ++j)
        {
            son_of_prime[primes[i] * primes[j]] = 1;
        }
    }

    for (int i = 0; i < T; ++i)
    {
        if (son_of_prime[numbers[i]])
            printf("Yes\n");
        else
            printf("No\n");
    }

    return 0;
}