UVa524 - Prime Ring Problem 枚举排列模板题

题意 给你n，用1~n排列成环，保证相邻两数之和为素数

思路 保证第一个数是1，然后用按序生成排列模板，每次先判断，满足条件否，不满足则剪枝。

注意：（1）两个样例输出间要有空行... 题目没说

（2）如果n为1的时候，我的结果是 1，交上去A了

#include <iostream>
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int MAX = 16 + 15 + 1;
int a[20];
int notsu[MAX];
int n;

void init()
{
int i,j;
for(i=2;i<MAX;i++)
{
for(j=2;j*j<=i;j++)
{
if(i % j == 0)
{
notsu[i] = 1;
break;
}
}
}
}

void out()
{
for(int i=1;i<=n;i++)
{
if(i != n)
printf("%d ",a[i]);
else
printf("%d\n",a[i]);
}
}

void dfs(int dep)
{
if(notsu[a[dep-1] + a[dep-2]])
return;
if(dep > n)
{
if(notsu[a[dep-1]+1])
return;
out();
return;
}
int i,j,tmp;
for(i=dep;i<=n;i++)
{
if(i > dep)
{
tmp = a[dep];
a[dep] = a[i];
for(j=i-1;j>dep;j--)
{
a[j+1] = a[j];
}
a[dep+1] = tmp;
}
dfs(dep+1);
if(i > dep)
{
tmp = a[dep];
for(j=dep;j<i;j++)
{
a[j] = a[j+1];
}
a[i] = tmp;
}
}
}

int main()
{
int i,j;
init();
int T = 0;
while(scanf("%d",&n) == 1 && n)
{
for(i=1;i<=n;i++)
{
a[i] = i;
}
if(T++)
puts("");
printf("Case %d:\n",T);
dfs(2);
}
return 0;
}