數獨 dfs

使用k(k/9,k%9)來進行每一個方格的搜索。
對於3*3宮,確定左定點:x=k/9/3*3,y=k%9/3*3.

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

int a[9][9]= {  {0,2,6,0,0,0,0,0,0},
                {0,0,0,5,0,2,0,0,4},
                {0,0,0,1,0,0,0,0,7},
                {0,3,0,0,2,0,1,8,0},
                {0,0,0,3,0,9,0,0,0},
                {0,5,4,0,1,0,0,7,0},
                {5,0,0,0,0,1,0,0,0},
                {6,0,0,9,0,7,0,0,0},
                {0,0,0,0,0,0,7,5,0}};
int vis[10][10];
int flag=0;

bool check(int k,int num){
    //行
    for(int j=0;j<9;j++)
        if(a[k/9][j]==num)
            return false;
    //列
    for(int i=0;i<9;i++)
        if(a[i][k%9]==num)
            return false;
    //3*3宮
    int x=k/9/3*3;//x爲3*3的左頂點橫座標 
    int y=k%9/3*3; //y爲3*3的左頂點豎座標
    for(int i=x;i<x+3;i++)
        for(int j=y;j<y+3;j++)
            if(a[i][j]==num)
                return false;
    return true; 
} 

void dfs(int k){
    if(flag)    return;
    if(k>=81){
        for(int i=0;i<9;i++){
            for(int j=0;j<9;j++)
                printf("%d ",a[i][j]);
            printf("\n");
        } 
        flag=1;
        return;
    }
    if(vis[k/9][k%9]){
        dfs(k+1);
        return;
    }
    for(int i=1;i<=9;i++){
        if(check(k,i)){
            a[k/9][k%9]=i;
            dfs(k+1);
            a[k/9][k%9]=0;
        } 
    } 
}

int main(){
    for(int i=0;i<9;i++)
        for(int j=0;j<9;j++){
            if(a[i][j]) vis[i][j]=1;
        }
    dfs(0);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章