素数环
题目描述
给定 n (n<=20) 把 1~n的n个数组成一个环,使得相邻的两个数和都是素数,如果不存在输出no solution
输入
一个n ( 2<=n<=20)
输出
输出这n个数 ,使得相邻的两个数都是素数,如果不存在输出no solution
样例输入
4
样例输出
1 2 3 4
提示
如果存在多组,请输出字典序最小的那个
分析:
DFS+剪枝
当N为奇数时,则数字1、2、…、N中数字为奇数的个数比偶数的个数多一个,必定存在两个奇数相邻的情况,奇数+奇数=偶数,不符合素数环的条件。
当N个数能组成素数环时,则数字1必定在素数环中的某个位置,将数字1移到素数环第一个位置,相当于将素数环旋转,依旧是一个素数环,因此可设素数环第一个数字为1,由此结合DFS可以推出字典序最小的素数环。
代码(C++)
#include<bits/stdc++.h>
using namespace std;
bool Prime(int n);
void DFS(int cnt,int n);
bool Flag=false;
bool Index[21];
bool flag[40];
int ans[21];
int main() {int n;cin>>n;fill(Index,Index+21,false);fill(flag,flag+40,false);for(int i=1; i<40; i++)flag[i]=Prime(i);ans[0]=1;Index[1]=true;if(n%2==0){DFS(1,n);}if(!Flag) {cout<<"no solution"<<endl;;}return 0;
}
bool Prime(int n) {if(n==1)return false;for(int i=2; i*i<=n; i++)if(n%i==0)return false;return true;
}
void DFS(int cnt,int n) {if(Flag)return;if(cnt==n&&flag[ans[cnt-1]+ans[0]]) {for(int i=0; i<cnt; i++)cout<<ans[i]<<" ";Flag=true;}for(int i=1; i<=n; i++) {if(!Index[i]&&flag[i+ans[cnt-1]]) {ans[cnt]=i;Index[i]=true;DFS(cnt+1,n);Index[i]=false;}}
}
代码(Java)
import java.util.Scanner;
public class Main {static boolean Flag=false;public static void main(String[] args) {Scanner cin = new Scanner(System.in);int n=cin.nextInt();boolean[] index=new boolean[21];boolean[] flag=new boolean[40];int[] ans=new int[21];int cnt=0;for (int i=1;i<flag.length;i++){flag[i]=Prime(i);}if(n%2==0){ans[0]=1;index[1]=true;DFS(1,n,ans,index,flag);}if(!Flag){System.out.println("no solution");}}public static boolean Prime(int n){if(n==1)return false;for(int i=2;i*i<=n;i++){if(n%i==0)return false;}return true;}public static void DFS(int cnt,int n,int[] ans,boolean[] index,boolean[] flag){if(Flag)return;if(cnt==n&&flag[ans[cnt-1]+ans[0]]){for(int i=0;i<cnt;i++)System.out.print(ans[i]+" ");Flag=true;}for(int i=1;i<=n;i++){if(!index[i]&&flag[i+ans[cnt-1]]){ans[cnt]=i;index[i]=true;DFS(cnt+1,n,ans,index,flag);index[i]=false;}}}
}