Description
有一个n*n矩阵A(0<=A[i][]j]<=9),现给出矩阵B,B[i][j]表示A[i][j]上下左右有多少个数字比A[i][j]大(0<=B[i][j]<=4),问满足矩阵B限制的A是否存在,如果存在输出一个方案,不存在则输出NO SOLUTION
Input
第一行为一整数n表示矩阵大小,之后为一n*n的矩阵B(1<=n<=3)
Output
如果存在满足B限制的A存在则输出一个方案,不存在则输出NO SOLUTION
Sample Input
3
1 2 1
1 2 1
1 1 0
Sample Output
1 2 3
1 4 5
1 6 7
Solution
dfs,一行行往下搜,一个重要剪枝是每次往A[i][j]放数字的时候就判断一下A[i-1][j]是否满足条件,不满足就不用往下搜了
Code
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
int n,flag,a[11],b[11];
int check(int x)
{
int res=0;
if(x>n&&a[x-n]>a[x])res++;
if(x+n<=n*n&&a[x+n]>a[x])res++;
if(x%n!=1&&a[x-1]>a[x])res++;
if(x%n!=0&&a[x+1]>a[x])res++;
if(res==b[x])return 1;
return 0;
}
void dfs(int x)
{
if(flag)return ;
if(x>n*n)
{
int temp=1;
for(int i=1;i<=n*n;i++)
if(!check(i))
{
temp=0;
break;
}
if(temp)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
printf("%d%c",a[(i-1)*n+j],j==n?'\n':' ');
flag=1;
}
return ;
}
for(int i=0;i<10;i++)
{
a[x]=i;
if(x<=n||check(x-n))
dfs(x+1);
}
}
int main()
{
while(~scanf("%d",&n))
{
for(int i=1;i<=n*n;i++)scanf("%d",&b[i]);
flag=0;
dfs(1);
if(!flag)printf("NO SOLUTION\n");
}
return 0;
}