藍橋杯訓練 C. 方格填數（第一周）

之前整理的題解連結 

#include<iostream>
#include<cstdio>
#include<cstring>
#define INF -100
using namespace std;
int Map[3][4];
int vis[11];
int cnt;
int disx[4]={0,-1,-1,-1};
int disy[4]={-1,-1,0,1};//左，左上，上，右上
int check(int x,int y,int n)
{
    for(int i=0;i<4;i++)
    {
        int xi=x+disx[i];
        int yi=y+disy[i];
        if(xi>=0&&xi<3&&yi>=0&&yi<4)
        {
            if(Map[xi][yi]==n-1||Map[xi][yi]==n+1)
                return 0;
        }
    }
    return 1;
}
void DFS(int x,int y)
{
    if(x==2&&y==3)
    {
        cnt++;
        return;
    }
    else if(y>=4)
        DFS(x+1,0);
    else
    {
        for(int i=0;i<=9;i++)
        {
            if(!vis[i]&&check(x,y,i)==1)
            {
                vis[i]=1;
                Map[x][y]=i;
                DFS(x,y+1);
                Map[x][y]=INF;
                vis[i]=0;
            }
        }
    }
}
int main()
{
    for(int i=0;i<3;i++)
    {
        for(int j=0;j<4;j++)
            Map[i][j]=INF;
    }
    memset(vis,0,sizeof(vis));
    cnt=0;
    DFS(0,1);
    cout<<cnt<<endl;
    return 0;
}