HDU 1016 Prime Ring Problem （素数筛+DFS）

题目链接

题意 ： 就是把n个数安排在环上，要求每两个相邻的数之和一定是素数，第一个数一定是1。输出所有可能的排列。

思路 : 先打个素数表。然后循环去搜。。。。。

//1016
#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std ;

bool vis[21];
int prime[42] ,cs[21];
int n ;

void get_prime()
{
    memset(prime,0,sizeof(prime));
    prime[1] = 1 ;
    for(int i = 2 ; i < 40 ; i ++)
    {
        if(prime[i] == 0)
        {
            for(int j = i*i ; j < 40 ; j += i)
                prime[j] = 1 ;
        }
    }
}

void DFS(int s,int cnt)
{
    if(cnt == n)
    {
        if(prime[cs[n]+1]) return ;
        for(int i = 1 ; i <= n - 1; i++)
            cout << cs[i] <<" " ;
        cout << cs[ n ]<<endl ;
        return ;
    }
    for(int i = 1 ; i <= n ; i++)
    {
        if(!vis[i] && !prime[s+i])
        {
            cs[++cnt] = i ;
            vis[i] = true ;
            DFS(i,cnt) ;
            cnt -- ;
            vis[i] = false ;
        }
    }
}
int main()
{
    int casee = 1 ;
    get_prime() ;
    while(  cin >> n )
    {
        memset(vis,false,sizeof(vis)) ;
        cout << "Case "<<casee++ <<":" << endl ;
        vis[1] = true ;
        cs[1] = 1;
        DFS(1,1) ;
        cout << endl ;
    }
    return 0 ;
}