HDOJ1016素数环 DFS

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include <math.h>
#include"queue"
#include"set"
#include"vector"
using namespace std;
int v[100];
bool isprime(int n)
{
    if(!v[n])
        return true;
    else 
        return false;
}
const int maxn=20+5;
int a[maxn];
int visit[maxn];
int n;
void put()
{
    int flag=1;
    for(int i=0;i<n;i++)
    {
        if(flag) flag=0;
        else printf(" ");
        printf("%d",a[i]);
    }
    printf("\n");
}
void dfs(int num,int def)
{
    a[def]=num;
    if(n-1==def)
    {
        put();
    }
    for(int i=1;i<=n;i++)
        if(!visit[i])
        {
            if((def==n-2&&isprime(i+a[0])&&isprime(i+num))) //当选中的是最后一个数时，他要与前一个数的和为素数并且要与第一个数的和都为素数
            {
                visit[i]=1;
                dfs(i,def+1);
                visit[i]=0;
            }
            else if(def>=0&&def<n-2&&isprime(i+num))  //当选中的是首尾中间的数 的时候他只要与前一个数的和为素数就可以了
            {
                visit[i]=1;
                dfs(i,def+1);
                visit[i]=0;
            }
        }
}
int main()
{
    memset(v,0,sizeof(v));
    v[0]=v[1]=1;
    for(int i=2;i<=(int)sqrt(100+0.5);i++)
        for(int j=i*i;j<=100;j+=i)
            if(!v[j]) v[j]=1;
    int Case=0;
    while(cin>>n)
    {
        printf("Case %d:\n",++Case);
        memset(visit,0,sizeof(visit));
        visit[1]=1;
        dfs(1,0);
        printf("\n");
    }
    return 0;
}