首页 > 代码库 > 素数环(dfs+回溯)
素数环(dfs+回溯)
题目描述:
输入正整数n,把整数1,2...n组成一个环,使得相邻两个数和为素数。输出时从整数1开始逆时针排列并且不能重复;
例样输入:
6
例样输出:
1 4 3 2 5 6
1 6 5 2 3 4
方法1:(生成测试法,会超时)
#include <bits/stdc++.h>
#define MAXN 100
using namespace std;
int isp[MAXN], a[MAXN];
void get_prime(void) //*****素数打表
{
memset(isp, 1, sizeof(isp));
isp[0]=isp[1]=0;
for(int i=2; i<MAXN; i++)
{
if(isp[i])
{
for(int j=2; j*i<MAXN; j++)
{
isp[i*j]=0;
}
}
}
}
int main(void)
{
std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
get_prime();
int n;
cin >> n;
for(int i=0; i<n; i++)
{
a[i]=i+1;
}
do
{
int flag=1;
for(int i=0; i<n; i++) //****逐个尝试
{
if(!isp[a[i]+a[(i+1)%n]]) //****判断合法性
{
flag=0;
break;
}
}
if(flag) //****如果合法则输出序列
{
for(int i=0; i<n; i++)
{
cout << a[i] << " ";
}
cout << endl;
}
}while(next_permutation(a+1, a+n)); //***1的位置不变
return 0;
}
方法2:(dfs+回溯)
代码:
#include <bits/stdc++.h>
#define MAXN 100
using namespace std;
int isp[MAXN], a[MAXN], vis[MAXN], n;
void get_prime(void) //****素数打表
{
memset(isp, 1, sizeof(isp));
isp[0]=isp[1]=0;
for(int i=2; i<MAXN; i++)
{
if(isp[i])
{
for(int j=2; i*j<MAXN; i++)
{
isp[i*j]=0;
}
}
}
}
void dfs(int cur)
{
if(cur==n && isp[a[0]+a[cur-1]]) //****递归边界
{
for(int i=0; i<n; i++) //***打印合法序列
cout << a[i] << " ";
cout << endl;
}
else
{
for(int i=0; i<n; i++) //***逐个尝试
{
if(!vis[i]&&isp[i+a[cur-1]]) //***i没用过且满足条件
{
a[cur]=i; //***存储当前序列
vis[i]=1; //****标记
dfs(cur+1); //***递归
vis[i]=0; //***去除标记
}
}
}
}
int main(void)
{
std::ios::sync_with_stdio(false),cin.fie(0),cout.fie(0);
get_prime();
cin >> n;
memset(vis, 0, sizeof(vis));
dfs(0);
return 0;
}
素数环(dfs+回溯)