【基础算法】素数环

题目描述

输入正整数n，把整数1，2，3，…，n组成一个环，使得相邻两个整数之和均为素数。小强同学看过这个题，笑了：呵呵，打表！
Mr. Wu为了阻止小强打表，决定这样：
把全部的解按字典序排序后，从1开始编号，依次输出指定编号的k组解。最后一行输出总的方案数。同一个素数环只算一次。

输入

第1行：2个整数，n(n<=18)和k(1<=k<=10)
第2行：共有k个从小到大排列的整数，表示要输出的解的编号。

输出

前k行，每行一组解，对应于一个输入
第k+1行：一个整数，表示总的方案数。

样例输入

(如果复制到控制台无换行，可以先粘贴到文本编辑器，再复制)

10 4
1 2 5 8

样例输出

1 2 3 4 7 6 5 8 9 10
1 2 3 4 7 10 9 8 5 6
1 2 3 8 5 6 7 10 9 4
1 2 3 10 9 8 5 6 7 4
96

提示

输入样例说明：

对1，2，...，10组成素数环。要输出字典序的第1，2，5，8等4组解

输出样例说明：

第1，2，5，8组解分别是样例中所列的4行。总共有96组解。

看见这个质数，就知道要先判断质数，预处理一下

bool pan[40];

for(int i=2;i<=39;i++)
		for(int j=2;j<=sqrt(i);j++)
			if(i%j==0)
			{pan[i]=true;break;}

为什么只处理到39呢？

看题目：n<=18  ！！

两个18加起来也就36，运行到39是避免越界，这样处理一下要快很多，不要每次都运行一遍

看到题目：把全部的解按字典序排序后，从1开始编号，依次输出指定编号的k组解。最后一行输出总的方案数。同一个素数环只算一次。

我第一反应就是：要用二维数组来存

但是又一想，最极限的时候，n=18可能有几万种可能，会不会越界啊，肯定原意是不会让你把每个可能都存进去的，又看到题目：“第2行：共有k个从小到大排列的整数，表示要输出的解的编号。”心里一喜，于是我用一个数组反复存，到了要求的行数后就输出，避免了越界。

然而因为对回溯不熟，还是有一些细节没改对

下面把错的都注释了：

#include<cstdio>
#include<cmath>
int n,k,a[20],s[20]={0,1},num,ii;
bool fg[20]={0,1},pan[40];//fg判断是否用过,pan判断素数,false是素数,true是质数
void print()
{
	ii++;
	for(int i=1;i<n;i++)
		printf("%d ",s[i]);
	printf("%d\n",s[n]);
}
void tt(int x)
{
	for(int i=2;i<=n;i++)
		if(!fg[i]&&!pan[i+s[x-1]])
		{
			s[x]=i;
			fg[i]=true;
			if(x==n/*写成i*/)
			{
				if(!pan[s[x]/*写成s[i]*/+s[1]]&&++num==a[ii+1]/*顺序写反：要先判断是否符合条件再计数*/) print();
			}
			tt(x+1);
			fg[i]=false;
		}
}
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=2;i<=39;i++)
		for(int j=2;j<=sqrt(i);j++)
			if(i%j==0)
			{pan[i]=true;break;}
	for(int i=1;i<=k;i++)
		scanf("%d",&a[i]);		
	tt(2);
	printf("%d",num);
	return 0;
}

终于改了一个多小时改对了。。。。。。