求大小为n的素数环（回溯法）

    

题目：将1~n的整数进行排列，使之两两相邻的数的和为素数，同时头尾相加也为素数，这样的排列成为素数环。输入一个n，请打印出所有的素数环。

input：

4

output：

2 1 4 3
2 3 4 1
3 2 1 4
3 4 1 2

思路：很经典的全排列算法，可以直接对1~n进行全排列，排列好后检查是否满足素数环的要求，但这样时间复杂度高达O（n！），我的电脑在当n=12时结果出的很慢了。这时可以使用回溯法，在排列数字的时候，我们检查一下与前一个数字的和是否为素数，如果不是直接回溯，当全部排列完毕后，检查一下头尾之和是否为一个素数，如果是，则将其打印。通过回溯法，即使n=16的情况下也能很快的打印出结果。此外由于频繁地判断素数，可以考虑用筛法筛出一个n*2大小的素数表。

代码如下：

import java.util.Scanner;

public class Main {
 static int a[],prime[];
 static boolean v[];
 static int n;
 public static void main(String[] args) {
      Scanner in = new Scanner(System.in);
      n = in.nextInt();
  
      prime_table(n*2);
      a = new int[n+1];
      v = new boolean[n+1];
  
      dfs(1);
 }
 
 public static void dfs(int i) {
      if(i==n+1) {
           if(is_prime(a[1]+a[n])) {//最后判断一下头尾之和是否为素数，是的话直接打印
                for(int j=1;j<=n;j++)
                     System.out.print(a[j]+" ");
                System.out.println();
           }
       return;
      }
      else {
           for(int j=1;j<=n;j++) {
                if(!v[j]) {
                     v[j] = true;
                     a[i] = j;
                     if(i!=1&&!is_prime(a[i]+a[i-1])) {//如果与前一个数的和不为素数，直接回溯
                          v[j] = false;
                          continue;
                     }
                     dfs(i+1);
                     v[j] = false;
                 }
           }
       }
   }
 
 public static boolean is_prime(int i) {
      return prime[i]!=1;
 }
 
 public static void prime_table(int max) {//筛出一个n*2的素数表，将里面的合数重置为1
      prime = new int[max+1];
      prime[0] = prime[1] = 1;
  
      for(int i=2;i<=max/2;i++) 
           for(int j=2;j<=i&&j*i<=max;j++) {
                if(prime[j]!=1) {
                     prime[j*i] = 1;
                     if(i%prime[j]==0)
                          break;
                 }
           }    
  }
}

如果你有更好的思路欢迎在评论区分享交流，感谢观看。