滨海做网站哪家最好/千峰培训多少钱

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;
}