方格填数 （简单DFS）

题目描述

如下的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;
}