蓝书-hdu1016 素数环(搜索-回溯)

题目

给一个n，将1-n的数围成一个圆环，

令环上的相邻两数之和为素数，输出所有排列可能

0<n<20

思路来源

蓝书

题解

感觉自己很多基础题不会啊，还是要巩固基础

却又不要沉迷在这些“简单”题上，虽然目前可能并不简单

 

直接暴搜肯定超时，所以考虑只搜那些可能对答案带来贡献的当前局面

所以就是下一步和当前步构成和为素数，才去搜下一步

写的比较丑啊QAQ

代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,cnt,step[20];
bool ok[45],vis[20];
void init()
{
	for(int i=2;i<45;++i)
	{
		for(int j=i*i;j<45;j+=i)
		ok[j]=1;
	}
} 
void dfs(int pos) 
{
	if(pos==n)
	{
		for(int i=0;i<n;++i)
		printf("%d%c",step[i],i==n-1?'\n':' ');
		return;
	}
	for(int i=2;i<=n;++i)
	{
		if(vis[i])continue;
		if(!ok[i+step[pos-1]]&&(pos<n-1||(pos==n-1&&!ok[1+i])))
		{
			step[pos]=i;
			vis[i]=1;
			dfs(pos+1);
			vis[i]=0;
		}
	}
}
int main()
{
	init();
	step[0]=1; 
	while(~scanf("%d",&n))
	{
		memset(vis,0,sizeof vis);
		printf("Case %d:\n",++cnt);
		dfs(1);
		puts("");
	}
	return 0;
}