题目描述：设计一个算法求解填字游戏问题，在3x3个方格的方阵中要填入0-9内的某9数字，每个方格填一个整数，使所有相邻的两个方格内的两个整数之和为素数。试求出所有满足条件的填法的个数。详见代码及注释，总体思路：DFS + 回溯。
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <iostream>
#include <cmath>
using namespace std;
int square[4][4];
int tag[10];
int ans = 0;
int S (int n) //判断是不是素数
{
for (int i = 2;i <= sqrt(n);i++)
{
if (n % i == 0)
return 0;
}
return 1;
}
bool judge(int x,int y,int num)
{
if (tag[num] == 1)
return false;
if (x - 1 >= 1) //判断当前位置上面的位置加上该数是不是素数
{
if (!S(square[x - 1][y] + num))
return false;
}
if (y - 1 >= 1) //判断当前位置左侧的位置加上该数是不是素数
{
if (!S(square[x][y - 1] + num))
return false;
}
return true;
}
void dfs(int x,int y)
{
if (x == 4)
{
ans++;
return;
}
for (int i = 0;i < 10;i++)
{
if (judge(x,y,i)) //如果判断条件成立
{
tag[i] = 1;//将已经用过的数标记为置1
square[x][y] = i;//将当前位置的数置为i
if (y!= 3)//如果没有到底，深度搜索
dfs(x,y+1);
else //否则，x+1
dfs(x+1,1);
square[x][y] = 0; //回溯
tag[i] = 0;
}
}
}
int main()
{
memset(tag,0,sizeof(tag));
dfs(1,1);
cout<<ans<<endl;
return 0;
}
