HDU 1016 素数环

题意：

一个环有n个点，将1-n填入这些点中，要求相邻两个点和为素数。输入n，输出所有可能的填法。

题意：利用dfs对n个数进行全排列，有符合题意的排列则输出。

#include <stdio.h>

int num[21];     //num数组存储的是数字的存储顺序 
int mark[21];    //mark数组是为了标记某个数字是否使用过,用过为0，没用过为1 
int n;    //共n个数     
//列出在题目规定范围内（0<n<20）所有可能出现的质数 
int prime_num[12] = {2,3,5,7,11,13,17,19,23,29,31,37};

//判断是否是质数，是返回1，不是返回0
int is_prime(int a)
{
    for(int i = 0; i < 12;i++)
        if(a == prime_num[i])
                return 1;
    return 0;
}

void print_num()    //打印从num[1]到num[n] 
{
    for(int i=1; i<n; i++)
        printf("%d ",num[i]);
    printf("%d",num[n]);
}
//深度优先搜索 ，pre和post表示搜索到的两个相邻的数
// flag表示搜索的深度，flag==n时结束 
int dfs(int pre, int post, int flag)    
{    
//如果相邻两个数相加不是素数，直接返回    
    if( !is_prime(pre+post) )
        return 0;    
    num[flag] = post;
    if(flag==n && is_prime(post+1))
    {
        print_num();
        printf("
");
        return 1;
    }
    //使用过了这个数字就标记为0
    mark[post] = 0;
    //遍历查找没有被用过的数 
    for(int i=2; i<=n; i++)
        if(mark[i]!=0 && dfs(post,i,flag+1))
                break;
    //标记位恢复原状
    mark[post] = 1;
    return 0;
}

int main() 
{
        //输出答案的次数 
    int count = 1;
    while(scanf("%d",&n) != EOF)
    {
        for(int i=1; i<=n; i++)
            mark[i] = 1;
        num[1] = 1;
        printf("Case %d:
",count++);
        if(n == 1)
                    printf("1
");
        for(int i=2; i<=n; i++)
            dfs(1, i, 2);
        printf("
");
    }
    return 0;
}