输入一个数n,然后在一个有n个节点的圆环上放1~n这n个数,求满足相邻两个数相加是素数的情况,并输出;
样例:
6
8
样例输出:
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(注意格式,两组之间有一个空行)
基本思路:
一看n的范围,必然是用dfs,所以就上吧,但是很快发现不行啊,时超限,那怎么办,剪枝啊,这个还是很好剪的,就是判断一下当前最新的两个是不是相加等于素数,如果不是,就直接return;
代码如下:
#include<iostream>
#include<iomanip>
#include<algorithm>
#include<queue>
#include<stack>
#include<list>
#include<set>
#include<map>
#include<vector>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
int prime[100+10];
int vis[30],temp[30];
int n;
void init()
{
memset(prime,0,sizeof(prime));
prime[1]=1;
int m=sqrt(50+0.5);
for(int i=2;i<=m;i++)
{
if(!prime[i])
{
for(int j=i+i;j<=50;j+=i)
{
prime[j]=1;
}
}
}
return;
}
void judge()
{
if(prime[temp[n]+temp[1]]) return;
for(int i=1;i<=n-1;i++)
{
if(prime[temp[i]+temp[i+1]])
return;
}
printf("1");
for(int i=2;i<=n;i++)
{
printf(" %d",temp[i]);
}
printf("\n");
}
void dfs(int cur)
{
if(cur==n+1)
{
judge();
return;
}
if(cur>2)
{
if(prime[temp[cur-1]+temp[cur-2]]) return;//剪枝
}
for(int i=2;i<=n;i++)//这个循环才是这个数组能取到每一种可能的原因,而不是下面的注释,那个注释那个地方会使程序爆炸,因为会一直存在某次递归到达不了cur+1;
{
if(!vis[i])
{
vis[i]=1;
temp[cur]=i;
dfs(cur+1);
vis[i]=0;
// dfs(cur);这个写法是错误的,之前没有弄明白;
}
}
return;
}
int main()
{
init();
int cnt=1;
while(scanf("%d",&n)==1)
{
printf("Case %d:\n",cnt++);
memset(vis,0,sizeof(vis));
temp[1]=1;
dfs(2);
printf("\n");
}
return 0;
}