洛谷P1218 [USACO1.5]特殊的质数肋骨 搜索

P1218 [USACO1.5]特殊的质数肋骨 Superprime Rib
题意 找出所有 n 位的十进制数要求其每一个前缀均为质数
搜索
加两个剪枝 ， 1、最高位有4种选择 ，可以选择 2 3 5 7
然后其他位只有 5 种选择 选 1 3 5 7 9
2、高位向低位枚举 这样 枚举的时候如果当前这个数已经不满足质数肋骨
那就直接 跳出，这是一个小优化 即一边做一边判断 是否为质数 这样就能剪掉很多

#include <cstdio>
#include <cmath>
#include <string>
#include <cstdlib>
#include <iomanip>
#include <iostream>
#include <algorithm>
#include <string>
using namespace std ;

int n ; 
int a[6] ;

bool is_prime(int x)     //  是质数  返回  true 
{
    if(x<=1) return false ; 
    for(int i=2;i*i<=x;i++) 
        if(x%i==0) return false ;
    return true ;
}

void dfs(int k,int sum ) 
{
    int x ; 
    for(int i=1;i<=5;i++) 
    {
        x = sum*10+a[i] ;
        if(is_prime(x)) 
        {
            if(k==n) printf("%d
",x) ; 
                else dfs(k+1,x) ;
        }
        
    }
    
}

int main() 
{
    scanf("%d",&n) ;
    a[ 1 ] = 1 ;
    a[ 2 ] = 3 ;
    a[ 3 ] = 5 ;
    a[ 4 ] = 7 ;
    a[ 5 ] = 9 ;
    int x ;
    for(int i = 1;i<=4;i++) 
    {
        x = a[ i ] ;
        if(i==1) x = 2 ;
        dfs(2,x) ;
    }
    
    return 0 ;
}