素数环问题 dfs+剪枝

1008: 素数环问题
时间限制: 1 Sec 内存限制: 128 MB
题目描述
把1到20这重新排列，使得排列后的序列A满足：
a. 任意相邻两个数之和是素数
b. 不存在满足条件a的序列B使得：A和B的前k（0 <= k <= 19）项相同且B的第k+1项比A的第k+1项小。
输入
没有输入。
输出
输出A，两个数字之间用一个空格隔开，第一个数字前面和最后一个数字后面没有空格。
提示
本题不提供输出样例。

#include <bits/stdc++.h>

using namespace std;
long long m,n;
long long ans;
int a[25],b[25];
bool flag=false;
bool pun(int t)
{
	for(int i=2;i<=sqrt(t);++i)
	{
		if(t%i==0)return false; 
	}
	return true;
}
void dfs(int u)
{
	if(flag)return;
	if(u>=21)
	{
		for(int i=1;i<20;++i)cout<<a[i]<<" ";
		cout<<a[20];
		flag=true;
		return;
	}
	for(int i=1;i<=20;++i)
	{
		if(b[i]==0)
		{
			if(i>=2&&pun(i+a[u-1])==false)continue;
			if(u==20&&pun(i+a[1])==false)continue;
			b[i]=1;
			a[u]=i;
			dfs(u+1);
			b[i]=0;
		}
	}
} 
int main()
{
	dfs(1);
	return 0;
}