经典的素数环问题,经典的DFS...然而我的DFS是用循环写的... 输入一个范围在1到19的整数n,将1~n按特定顺序围成一个环,若这个环每相邻两个数的和都是素数,那这就是一个素数环, 输出每个整数的所有素数环序列,并在每例后面输出一个空行
深搜(DFS),(或者叫回溯算法) 从1开始,搜索下一个可以与之相加得到素数的数,每搜一层都从第一个结果继续往下深挖... 挖到不再能搜出新结果为止, 此时判断1~n的所有数是否都已使用,末尾结果与1是否能成环, 如果都是,那就得到了一条答案路径,输出再回溯; 否则,直接回溯。(因此无论是否得到答案都要回溯)
搜索的筛选条件之一是与被搜数的和必须是素数,因此需要用到素数判定算法。 本代码用埃氏筛法打素数表(埃氏筛是一种可快速得到n以内的素数表的算法,具体步骤可参考百度百科), 本人在摸本题代码的过程当中提炼出一个埃氏筛素数打表通用函数(模板),以后有需要打素数表的题目都可以直接使用(见下方AC代码)
本人很奇葩地用循环实现了DFS... 具体实现是, 用二维数组dfs[][]来存储搜索树, 每行的第一个数表示当前层的选择数,后面的数则是这一层可选择的数的序列, 比如搜索6的素数环, 第一层只能选择1,那么dfs的第一行数据是1,1 第二层可以选择2,4,6,那么第二行数据是3,2,4,6(有3个选择,分别是2,4,6),以此类推, 在搜索过程中,dfs数组中的数会随着搜索过程而变化 搜索的筛选条件除了要与被搜数的和是素数,还不能和前面每一层的被搜数重复,因此要一个标记数组vis来辅助除重, 当下一层一个选择都搜索不出的时候,检查答案是否产生,然后回溯, 具体的回溯操作是, 回到上一个选择数>=2的那一层,(类似于迷宫题中“传送回上一个走过的岔路口”) 同时取消回溯点后面的一路上所有数的visited标记, 并进行“封路”操作(回溯到的那一层的选择数-1,删除第一个可选择的数(即被搜数),后面的可选择数向前覆盖), 然后再从此时这一层的最靠左的可选择数开始,重复以上搜索和回溯。 在这套搜索逻辑下,每一层的被搜数总是排在最靠左的位置(即dfs数组中每一行的第二个数,第一个数是这一行的选择数) 因此在需要输出答案路径的时候,按序输出数组dfs中每一行的第二个数即可
注意这题每个case后面还要跟一个空行
#include<iostream>
using namespace std;
int PrimeFlag[50];
void makePrimeFlag(int n)//用埃氏筛法打出n以内的素数表(下标标记版)的函数,配合名为PrimeFlag的全局数组使用,可作为模板
{
for (int i = 2; i <= n; i++)
PrimeFlag[i] = 1;
int minflag = 0, maxflag = n;
while (maxflag >= minflag * minflag)
{
for (int i = minflag + 1; i <= maxflag; i++)
{
if (PrimeFlag[i] == 1)
{
minflag = i;
break;
}
}
for (int i = minflag + 1; i <= maxflag; i++)
{
if (i%minflag == 0)
PrimeFlag[i] = 0;
}
for (int i = n - 1; i >= minflag; i--)
{
if (PrimeFlag[i] == 1)
{
maxflag = i;
break;
}
}
}
}
int main()
{
makePrimeFlag(50);
int n;
int t = 0;
int dfs[30][20];
int vis[20];
int location;
int layer;
int direction;
while (cin >> n)
{
t++;
cout << "Case " << t << ":" << endl;
dfs[1][0] = 1;
dfs[1][1] = 1;
for (int i = 0; i < 20; i++)
vis[i] = 0;
vis[1] = 1;
//DFS
layer = 1;
while (true)
{
direction = 0;
location = dfs[layer][1];
for (int i = 1; i <= n; i++)
{
if (i != location && PrimeFlag[i + location] == 1 && vis[i] == 0)
{
direction++;
dfs[layer + 1][direction] = i;
if(direction==1)
vis[i] = 1;
}
}
dfs[layer + 1][0] = direction;
if (dfs[layer + 1][0] == 0)
{
if (PrimeFlag[location + 1] == 1 && layer == n)//符合答案时,先输出路径再回溯
{
for (int i = 1; i <= n; i++)
{
cout << dfs[i][1];
if (i < n)
cout << " ";
}
cout << endl;
}
//进行回溯、封路、去重+换重操作
while (dfs[layer][0] <= 1)
{
if (layer == 0)
break;
vis[dfs[layer][1]] = 0;
layer--;
}
if (layer == 0)
break;
else
{
vis[dfs[layer][1]] = 0;
for (int i = 1; i <= dfs[layer][0]; i++)
{
dfs[layer][i] = dfs[layer][i + 1];
}
vis[dfs[layer][1]] = 1;
dfs[layer][0]--;
layer--;
}
}
layer++;
}
cout << endl;
}
return 0;
}