素数圈 DFS

素数圈 DFS
原题链接https://vjudge.net/contest/343404#problem/G


没啥可说 上注释

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
long long book[10005];//标记判断是否可以使用
long long result[10005];//记录使用过的数字
long long n;
long long sum;
long long prime(long long n)//判断是否为素数
{
	if(n<=1)//去除特殊情况
		return 0;
	long long i;
	for(i=2; i<=sqrt(n); i++)//用素数筛进行判断
	{
		if(n%i==0)
		{
			return 0;
		}
	}
	return 1;
}
long long check(long long i,long long step)//判断数是否可以使用
{
	if((book[i]==0)&&prime(i+result[step-1])==1)//判断数是否使用过，是否为素数
	{
		if(step==n-1)//如果为最后一个数要判断和第一个数相加是否为素数
		{
			if(prime(i+result[0])==0)
				return 0;
		}
		return 1;
	}
	return 0;
}
void dfs(int step)
{
	if(step==n)//判断已经使用过的数的个数
	{
		long long a;
		printf("%lld",result[0]);
		for(a=1; a<n; a++)
		{
			printf(" %lld",result[a]);
		}
		printf("\n");
		return ;
	}
	long long i;
	for(i=2; i<=n; i++)//从第二个数开始寻找可以使用的数
	{
		if(check(i,step))//判断数是否可以使用
		{
			book[i]=1;//记录
			result[step]=i;//下一位数更新为使用的数
			dfs(step+1);//次数加一向下传递
			book[i]=0;//如果死路还原
		}
	}
}
int main()
{
	sum=1;//记录多组此数
	while(~scanf("%lld",&n))
	{
		memset(result,0,sizeof(result));//格式化
		memset(book,0,sizeof(book));//格式化
		result[0]=1;//输入第一个数为1
		printf("Case %lld:\n",sum);//输出格式
		dfs(1);//进行dfs
		sum++；//多组次数
		printf("\n");
	}
	return 0;
}