【USACO】洛谷1218 特殊的质数肋骨 Superprime Rib

题目描述

农民约翰的母牛总是产生最好的肋骨。你能通过农民约翰和美国农业部标记在每根肋骨上的数字认出它们。农民约翰确定他卖给买方的是真正的质数肋骨,是因为从右边开始切下肋骨,每次还剩下的肋骨上的数字都组成一个质数,举例来说:
7 3 3 1 全部肋骨上的数字 7331是质数;三根肋骨 733是质数;二根肋骨 73 是质数;当然,最后一根肋骨 7 也是质数。
7331 被叫做长度 4 的特殊质数。写一个程序对给定的肋骨的数目 N (1<=N<=8),求出所有的特殊质数。数字1不被看作一个质数。
输入输出格式 输入格式：

单独的一行包含N。

输出格式：

按顺序输出长度为 N 的特殊质数,每行一个。

很容易发现，最高位的数字只能是2，3，5，7，其他位的数字只能是1，3，7，9【考虑这一位结尾的那个数】。
dfs枚举即可。

#include<cstdio>
#include<cstring>
#include<cmath>
int n,tot,a[]={2,3,5,7},b[]={1,3,7,9},ans[1000000];
bool is(int x)
{
    int y=sqrt(x+0.5);
    for (int i=2;i<=y;i++)
      if (x%i==0) return 0;
    return 1;
}
void dfs(int now,int k)
{
    int i,j,x,y,z;
    if (!is(now)) return;
    if (k==-1)
    {
        ans[++tot]=now;
        return;
    }
    if (k==n-1)
      for (i=0;i<4;i++)
        dfs(now*10+a[i],k-1);
    else
      for (i=0;i<4;i++)
        dfs(now*10+b[i],k-1);
}
int main()
{
    int i,j,k,x,y,z;
    scanf("%d",&n);
    dfs(0,n-1);
    for (i=1;i<=tot;i++)
      printf("%d\n",ans[i]);
}