算法导论3 B 有趣的素数(dfs)

题目：

题目描述

素数被广泛地应用于密码学中，所谓的公钥就是将想要传递的信息在编码时加入砠数，编码之后传给收信人，任何人收到此信息之后，若没有此收信人所拥有的秘钥，则在解密的过程中将会因为分解质因数过久而无法破解信息，可见素数在密码学中的重要性。
现在给你n（2<=n<=16）个正整数1,2,3...n，你的任务是把这n个正整数组成一个环，使得任意相邻的两个整数之和为一个素数，输出有多少种合法方案。

输入

多组输入数据，每组数据只有一个正整数n（2<=n<=16）代表有n个正整数 1,2,3...n

输出

对每组数据，输出一个整数，代表有多少种不同的可行方案数。

样例输入 Copy

6

8

样例输出 Copy

2

4

解析：

这道题第一眼可能会把他当成dp题，很容易让人联想能否从第i个素数环中推导出第i+1个素数环的个数。

然而再仔细看就会发现这是一道dfs题。

深度优先遍历所有排列情况，找出满足要求的个数。

代码：

#include<iostream>
#include<bitset>
using namespace std;
const int N = 100;
bitset<N> prime;//埃氏筛 
bitset<17> vis;//记录这个数字是否被放入环中 
int n,ans;//题目输入的n，答案ans
int t[17]; //记录一下这个环，方便环的首位相加是否是素数的判断
void dfs(int x)
{
	if(x==n)
	{
		if(!prime[t[0]+t[x-1]])//别忘了是一个环，首尾相连也要是素数 
			ans++;
		return;
	}
	else
	{
		for(int i=1;i<=n;i++)
		{
			if(!vis[i])//没有被加入到环里
			{
				if(!prime[t[x-1]+i])//和前面的整数相加是素数
				{
					vis[i] = true;
					t[x] = i;
					dfs(x+1);
					vis[i] = false;
				}
			} 
		}
	}
}
int main()
{
	prime[0] = true,prime[1] = true;//小于100的数量级没必要用埃氏筛，这里是怕手生写一遍练手
	for(int i=2;i<N;i++)
		if(!prime[i])
			for(int j=2*i;j<N;j+=i)
				prime[j] = true;
	while(cin>>n)
	{
		vis[1] = true;//默认1开头(实际上没有开头 )
		for(int i=2;i<=n;i++)
			vis[i] = false;
		ans = 0;
		t[0] = 1;//默认1开头 
		dfs(1);
		cout<<ans<<'\n';
	}
}