题目:将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;
}
}
}
}
如果你有更好的思路欢迎在评论区分享交流,感谢观看。