PTA 吸血鬼素数

题目作者 胡伟平          单位 广西科技大学

吸血鬼素数是指这样的数：如果两个素数位数相同，并且乘积正好与这两个素数的组成数字相同，那么这两个素数都被称为吸血鬼素数

例如，117067=167×701，左右两边都由一个“0”，两个“1”，一个“6”，两个“7”组成，而167和701都是素数，那么167和701都是3位吸血鬼素数。

请判断给定的一组数，是否是吸血鬼素数。

输入格式:

有多组测试数据，第一行输入测试数据的组数T，随后输入T个正整数N。

0 < T, N < 10000

输出格式:

每一行内，对一组输入如果是吸血鬼素数则输出“YES”，否则输出“NO”。

 

#include <iostream>
using namespace std;

int t, n, m, c, w, k, a[10], sum, p[1300];    //p存质数，一万以内约有1250个
bool st[9999];            //st存i是否是更小的值的倍数。true表示可以，false表示为质数。

void zz(int l)  {          //欧拉筛判质数,不用会超时
    for (int i=2; i <= l; i++)  {
        if (!st[i]) p[sum++]=i;
        for (int j=0; p[j]<=l/i; j++)  {
            st[i*p[j]]=true;
            if (i%p[j]==0) break;
        }
    }
}

int ww(int n)  {            //判位数
    string s;
    s=to_string(n);
    return s.size();
}

void cc(int b, int d)  {    //操作，计算每个数有多少个
    while (b)  {
        a[b%10]+=d;
        b/=10;
    }
}

bool pp()  {                //如果a都是0则为真
    for (int i=0; i<10; i++)
        if (a[i]) return false;
    return true;
}

int main ()  {
    zz(9999);
    cin >> t;
    while (cin >> n, t--)  {
        if (st[n])  {        //如果n不是质数，输出NO并跳过
            cout << "NO" <<endl;
            continue;
        }

        k=1;
        w=ww(n);
        for (int i=0; i<w; i++) k*=10;

        int l;
        cc(n,1);
        for (l=0; p[l]<k; l++)
            if (p[i]>k/10)  {
                c=n*p[l];
                cc (p[l],1), cc(c,-1);
                if (pp())  break;
                cc (p[l],-1), cc(c,1);    //重置，防止影响下一个p[l]
            }

        if (p[l]<k)  cout << "YES" << endl;
        else cout << "NO" << endl;
        for (int i=0; i<10; i++)  a[i]=0;        //重置数组a
    }
    return 0;
}