递归与回溯 HDOJ 2553 N皇后问题 1016 Prime Ring Problem

#include <iostream>
#include <cmath>

using namespace std;

int f[15], cnt, n;

int isplace(int m)
{
for (int i = 1; i < m; ++i)
{
// 判断是否和已有棋子冲突，只需判断列和对角线即可
if (f[m] == f[i] || abs(m - i) == abs(f[m] - f[i]))
return 0;
}
return 1;
}

//m是当前正在放第几个棋子，并把这个棋子放在第m行
void dfs(int m)
{
// 递归出口
if (m == n + 1) {
cnt ++;
return ;
}
// 从第一列枚举到最后一列
for (int i = 1; i <= n; ++ i)
{
f[m] = i; // 这一步在枚举的同时也包含了回溯
// 若可以放置，则放置下一个棋子
if (isplace(m)) {
dfs(m + 1);
}
}
return ;
}

int main(int argc, char const *argv[])
{
int i,j,ans[15];
//先打表，不然会超时
for (int i = 1; i <= 10; ++ i)
{
cnt = 0;
n = i;
dfs(1);
ans[i] = cnt;
}
int t;
while (cin >> t, t){
cout << ans[t] << endl;
}

return 0;
}

#include <iostream>
#include <cmath>
#include <cstring>
#include <cstdio>

using namespace std;

int isok[25], f[25];

// 判断一个数是否是素数
int IsPrime(int n)
{
if (n == 2) return 1;
if (n % 2 == 0) return 0;
for (int i = 3; i <= sqrt(n); i += 2)
{
if (n % i == 0)
return 0;
}
return 1;
}

void dfs(int cur, int n)
{
// 递归出口
if (cur == n + 1) {
/* 需要注意的是，所有数字放好之后还要判断首尾两个
数的和是否是素数 */
if (IsPrime(f
+ f[1]) ) {
cout << f[1];
for (int i = 2; i <= n; ++i)
{
cout << " " << f[i];
}
cout << endl;
}
return ;
}
for (int i = 2; i <= n; ++i)
{
if ( isok[i] && IsPrime(i + f[cur - 1]) )
{
f[cur] = i;
// 标记当前数字已被使用
isok[i] = 0;
dfs(cur + 1, n);
// 回溯，重新标记当前数字未被使用
isok[i] = 1;
}
}
return ;
}

int main(int argc, char const *argv[])
{
int n, cnt = 1;
f[1] = 1;
while (cin >> n) {
memset(isok,-1,sizeof(isok));
isok[1] = 0;
printf("Case %d:\n",cnt ++);
dfs(2,n);
cout << endl;
}

return 0;
}