HDU 1016 素数环问题

题目大意:

给定1-n这n个数，组成以1开头的素数环，保证相邻两个数相加均为素数

题目用dfs搜索再回溯，这样碰到不成立的立刻退出递归，就减少了很多步骤，不然暴力来就是n！次复杂度，肯定是超时的

每次添入数据都要判断是否相邻数相加为素数，所以我们可以提前打个素数表，这样使自己判断素数更加方便

 #include <cstdio>
 #include <cstring>

 using namespace std;

 #define N 22
 int n , a[N] , visit[N] , prime[];

 void prime_table()
 {
     memset(prime,,sizeof(prime));
     for(int i = ;i<;i++){
         if(!prime[i]){
             for(int j=i+i;j<;j+=i)
                 prime[j] = ;
         }
     }
 }

 void dfs(int x,int cnt)
 {
     visit[x] = ;
     if(cnt == n && !prime[a[]+a[cnt-]]){
         printf("%d",a[]);
         for(int i=;i<cnt;i++)
             printf(" %d",a[i]);
         printf("\n");
     }

     for(int i=;i<=n;i++){
         if(!visit[i]&& !prime[a[cnt-]+i]){
             a[cnt] = i;
             dfs(i,cnt+);
             visit[i]=;//回溯
         }
     }
 }

 int main()
 {
     int t = ;
     prime_table();
     while(~scanf("%d",&n))
     {
         printf("Case %d:\n",t);

         memset(visit,,sizeof(visit));
         a[] = ;
         dfs(,);
         //这里一组数据做完记得换行，题目要求
         printf("\n");
         t++;
     }
 }