问题描述:将从1到n这n个整数围成一个圆环,若其中任意2个相邻的数字相加,结果均为素数,那么这个环就成为素数环,输入n,如果能构成素数环则输出该n下的所有情况。原题地址http://acm.hdu.edu.cn/showproblem.php?pid=1016
Sample Input
6 8
Sample Output
Case 1: 1 4 3 2 5 6 1 6 5 2 3 4 Case 2: 1 2 3 8 5 6 7 4 1 2 5 8 3 4 7 6 1 4 7 6 5 8 3 2 1 6 7 4 3 8 5 2
额、、、、、先来说说吧,这题是到很老(貌似最早96年出现的)却很经典的问题。对于才接触深度优先搜索(DFS)的朋友来说是个不错的练习题,先来说说我(本人才入ACM坑,新手勿喷)对于这题的思路。
我们举个特例,比如6,我们把每个能做相邻元素的数字连接起来(额,图好丑)构成一个完全图。
在这个完全图中选出能使两个元素相加为素数的连线。
如图,如果这个图可以一笔画从1出发,走遍所以定点并且回到1(一条边最多走1次,可以有没走过的边),那么这种情况就能构成素数环,否则不能。在n=6的情况下有两种情况(1,4,3,2,5,6和1,6,5,2,3,4)因此我写代码就是遵从这连个条件写的(一,把和为素数的连接起来。二,如果能从1出发,边最多走一次的走过所有定点并且回到1.)哎~废话这么久,翠花~上代码!
#include <iostream>
#include<cmath>
using namespace std;
int n;
int a[21]={0};//把从1出发符合和为素数的顶点存在里面,从a[1]放1,a[2...n]放其他数。
int b[21]={0};//记录是否被访问过
int pi(int n)//素数判断的函数
{
int i;
if(n==1)
return 0;
for(i=2;i<=(int)sqrt((double)n);i++){
if(n%i==0)
return 0;
}
return 1;
}
void dfs(int cur)
{ int i;
if(cur>n&&(pi(a[n]+1)))/*如果走过的结点符合例如红线图的方式,也就是都走相连接顶点和为素数的线,cur才能大于n,<span style="font-family: Arial, Helvetica, sans-serif;">并且和1的和为素数,</span>那么就把存在a【n】的走过的顶点记录输出*/
{
for(i=1;i<n;i++)
{
cout<<a[i]<<" ";
}
cout<<a[n];
cout<<endl;
}
else
{
for(i=2;i<=n;i++)//从2开始,也就是先1连接2
{
if(b[i]==0&&pi(i+a[cur-1]))
//这里就是上文所描述的条件之一(判断这个数是否与上个数的和为素数)。以及是否被遍历过。
{
a[cur]=i;//符合了条件就把走过的定点放进数组
b[i]=1;//标记该节点被访问过
dfs(cur+1);//访问再下个节点
b[i]=0;//清除访问记录
}
}
}
}
int main(int argc, char *argv[])
{
int i=1;
while(cin>>n){
a[1]=1;
b[1]=true;
cout<<"Case "<<i++<<":"<<endl;
dfs(2);//1存放在a[1],所以从2开始。
cout<<endl;
}
return 0;
}
呼,总算写完了。本人是才进ACM的新人,这是我发的第一篇文章,请各位多多指教。
这个代码比较耗时,800多ms,看有的人200ms,各位大神们,有什么优化代码的方案呢?