ProgramAlgTrain/20240823/十四届蓝桥比赛/数独填数.cpp

#include<bits/stdc++.h>
using namespace std;

int m[10][10];
bool h_is_used[10][10],l_is_used[10][10],t_is_used[4][4][10];

#ifdef OI
#define DB(code){code}
#define PV(v){cout<<#v<<" : "<<(v)<<endl;}
#else
#define DB(code)
#define PV(v)
#endif

void dfs(int x,int y);
optional<pair<int,int>> find_next(int x,int y);

int main(){
    for(int i=1;i<=9;i++){
        string str;
        getline(cin,str);
        for(int j=1;j<=9;j++){
            if(str[j-1]=='.'){
                m[i][j]=-1;
            }else{
                m[i][j]=str[j-1]-'0';
                h_is_used[i][m[i][j]]=true;
                l_is_used[j][m[i][j]]=true;
                t_is_used[i/3+1][j/3+1][m[i][j]]=true;
            }
        }
    }
    DB(
        for(int i=1;i<=9;i++){
            for(int j=1;j<=9;j++){
                cout<<m[i][j]<<"\t";
            }
            cout<<endl;
        }
    )
    auto find = find_next(1, 1);
    if(find.has_value()==false){
        for(int i=1;i<=9;i++){
            for(int j=1;j<=9;j++){
                cout<<m[i][j];
            }
            cout<<endl;
        }
        exit(0);
    }
    dfs(find->first, find->second);
}

optional<pair<int,int>> find_next(int x,int y){
    for (int j=y+1; j<=9; j++) {
        if(m[x][j]==-1){
            return make_optional<pair<int,int>>({x,j});
        }
    }
    for(int i=x+1;i<=9;i++){
        for(int j=1;j<=9;j++){
            if(m[i][j]==-1){
                return make_optional<pair<int,int>>({i,j});
            }
        }
    }
    return nullopt;
}   

void dfs(int x,int y){
    DB(PV(x)PV(y)cout<<endl;)
    auto find = find_next(x, y);
    if(find.has_value()==false){
        for(int i=1;i<=9;i++){
            for(int j=1;j<=9;j++){
                cout<<m[i][j];
            }
            cout<<endl;
        }
        exit(0);
    }
    for(int i=1;i<=9;i++){
        if(h_is_used[x][i]||l_is_used[y][i]||t_is_used[x/3+1][y/3+1][i]){
            continue;
        }
        h_is_used[x][i]=true,
        l_is_used[y][i]=true,
        t_is_used[x/3+1][y/3+1][i]=true;
        m[x][y]=i;

        dfs(find->first, find->second);

        h_is_used[x][i]=false;
        l_is_used[y][i]=false;
        t_is_used[x/3+1][y/3+1][i]=false;
        m[x][y]=-1;
    }
}