题目描述
如下的10个格子
+--+--+--+
| | | |
+--+--+--+--+
| | | | |
+--+--+--+--+
| | | |
+--+--+--+
填入0~9的数字。要求:连续的两个数字不能相邻。
(左右、上下、对角都算相邻)
一共有多少种可能的填数方案?
题解
这是一道最简单的dfs题,添加判断函数即可
# include <cstdio>
# include <cstring>
# include <cmath>
# include <algorithm>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 10;
int map[5][5];
int f[10];
int ans;
int ok(int r, int c, int val)
{
if(r==0 && c==1) return 1;
else if(r==0) return abs(val-map[r][c-1])>1;
else if(r==1 && c==0) return abs(val-map[r-1][c+1])>1;
else if(r==1 && c==1) return abs(val-map[r][c-1])>1&&abs(val-map[r-1][c])>1&&abs(val-map[r-1][c+1])>1;
else if(r==1 && c==2) return abs(val-map[r][c-1])>1&&abs(val-map[r-1][c-1])>1&&abs(val-map[r-1][c])>1&&abs(val-map[r-1][c+1])>1;
else if(r==1) return abs(val-map[r][c-1])>1&&abs(val-map[r-1][c-1])>1&&abs(val-map[r-1][c])>1;
else if(r==2 && c==0) return abs(val-map[r-1][c])>1&&abs(val-map[r-1][c+1])>1;
else return abs(val-map[r][c-1])>1&&abs(val-map[r-1][c-1])>1&&abs(val-map[r-1][c])>1&&abs(val-map[r-1][c+1])>1;
}
void dfs(int r, int c)
{
if(r==2 && c==3)
{
ans++;
return;
}
for(int i=0;i<=9;i++)
{
if(!f[i] && ok(r, c, i))
{
map[r][c] = i;
f[i] = 1;
dfs(r+(c+1)/4, (c+1)%4);
f[i] = 0;
}
}
}
int main()
{
ans = 0;
memset(f,0,sizeof(f));
dfs(0,1);
printf("%d\n",ans);
return 0;
}
如果觉得判断过于冗长,可以在初始化数组时加上(0,0)和(2,3)的值,保证和其他格子里的数值相差值大于1即可
# include <cstdio>
# include <cmath>
using namespace std;
int num[3][4];
int sum = 0;
bool isFilled[10];
bool isLegal(int r, int c, int val)
{
if( r == 0 && c == 0 )
return true;
else if( r == 0 )
return abs(val-num[r][c-1]) > 1;
else if( c == 0 )
return abs(val-num[r-1][c]) > 1 && abs(val-num[r-1][c+1])>1;
else if( c != 3 )
return abs(val-num[r][c-1]) > 1 && abs(val-num[r-1][c]) > 1 && abs(val-num[r-1][c-1])>1 && abs(val-num[r-1][c+1])>1;
else
return abs(val-num[r][c-1]) > 1 && abs(val-num[r-1][c]) > 1 && abs(val-num[r-1][c-1])>1;
}
void dfs(int r, int c)
{
if( r == 2 && c == 3 )
{
sum++;
return;
}
for(int i=0;i<=9;i++)
{
if( !isFilled[i] && isLegal(r, c, i) )
{
num[r][c] = i;
isFilled[i] = true;
dfs( r + (c+1)/4, (c+1)%4 );
isFilled[i] = false;
}
}
}
int main()
{
num[0][0] = -10;
num[2][3] = -10;
for(int i=0;i<10;i++)
isFilled[i] = false;
dfs( 0, 1 );
printf("%d\n", sum);
return 0;
}