SGU 125 Shtirlits（dfs）

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