ACM 66. [HAOI2004模拟] 数列问题(水dfs)

66. [HAOI2004模拟] 数列问题

★☆   输入文件： dfs3.in   输出文件： dfs3.out    简单对比
时间限制：1 s   内存限制：128 MB

问题描述
试编程将 1 至 N （ N ≤ 15 ）的自然数序列 1 ， 2 ， … ， N 重新排列，使任意相邻两数之和为素数。例如 N=3 时有两种排列方案 123 、 321 满足要求。

【输入格式】

输入文件：dfs3.in

第一行：一个整数n(1<=n<=15)

【输出格式】

输出文件：dfs3.out

输出若干行，每行为一种排列方案（排列方案按字典序排列， 相邻数字之间用空格分隔） ），最后一行输出排列方案总数。

【输入样例】

输入文件名：dfs3.in

3

输出文件名：dfs3.out

1 2 3
3 2 1
2

水dfs

#include <iostream>
#include <cstdio>

using namespace std;

#define MAX_N 16

int n;
int sel[MAX_N];
bool used[MAX_N];
int cnt=0;

bool IsPrime(int n)
{
	int i=2;

	while(i*i<=n)
	{
		if(n%i==0) return false;
		i++;
	}

	return true;
}

void dfs(int cur)
{
	if(cur==n)
	{
		for(int i=0;i<n-1;i++)
			printf("%d ",sel[i]);
		printf("%d\n",sel[n-1]);
		cnt++;
		return;
	}

	for(int i=1;i<=n;i++)
	{
		if(!used[i] && IsPrime(sel[cur-1]+i))
		{
			sel[cur]=i;
			used[i]=true;
			dfs(cur+1);
			used[i]=false;
		}
	}
}

int main()
{
	freopen("dfs3.in","r",stdin);
	freopen("dfs3.out","w",stdout);

	scanf("%d",&n);

	for(int i=1;i<=n;i++)
	{
		sel[0]=i;
		used[i]=true;
		dfs(1);
		used[i]=false;
	}
	printf("%d\n",cnt);

	return 0;
}