dfs—素數環

題目:
在這裏插入圖片描述

import java.util.Scanner;

public class 素數環 {
	public static void main(String[] args) {
		Scanner sc=new Scanner(System.in);
		int n=sc.nextInt();
		int [] r=new int[n];
		r[0]=1;
		dfs(n,r,1);
	}

	private static void dfs(int n, int[] r, int cur) {
		//填到末尾了還有首尾相加爲素數
		if (cur==n&&isP(r[0]+r[n-1])) {
			print(r);
			return;
		}
		for (int i = 2; i <= n; i++) {//嘗試用每個數字填到cur這個位置
			//如果滿足條件,即r中沒有i這個數,且和上一個數爲素數
			if (check(r,i,cur)) {
				r[cur]=i;//嘗試將i這個數放在cur位置上,往前走一步
				dfs(n,r,cur+1);
				r[cur]=0;//回溯
			}
		}
	}

	private static void print(int[] r) {
		for (int i = 0; i < r.length; i++) {
			System.out.print(r[i]+(i==r.length-1?"":" "));
		}
		System.out.println();
	}

	private static boolean check(int[] r, int i, int cur) {
		for(int e:r) {
			if(e==i||!isP(r[cur-1]+i))
				return false;
		}
		return true;
	}

	private static boolean isP(int k) {
		for (int i = 2; i*i<=k; i++) {
			if (k%i==0) {
				return false;
			}
		}
		return true;
	}
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章