DFS - POJ 2676 - Sudoku

最新推荐文章于 2020-06-12 21:07:51 发布

njuptACMcxk

最新推荐文章于 2020-06-12 21:07:51 发布

阅读量144

点赞数

分类专栏： DFS 文章标签：算法 dfs 搜索 acm竞赛

本文链接：https://blog.csdn.net/njuptACMcxk/article/details/106277385

版权

DFS 专栏收录该内容

22 篇文章 0 订阅

订阅专栏

DFS - POJ 2676 - Sudoku

$给定一个 9 \times 9 的方阵进行数独游戏。$
在这里插入图片描述

$要求每一行，每一列，每一个小九宫格中均不能有重复数字。$

$T 组测试用例，每组包括一个 9 \times 9 的数字方阵，^{'} 0^{'} 表示空位。$

$保证有解，输出结果。$

Sample Input

Sample Output

Time limit： 2000 ms
Memory limit： 65536 kB

分析：

$暴力 d f s 每一个空位能够枚举的数字，找到一组可行解就返回。$

$剪枝方案：$

$① 、优化搜索顺序：优先搜索分支较少的节点 (能够选择的数字较少的空位) 。$

$② 、可行性剪枝：行、列、小九宫格内不能存在重复数字。$

具体落实：

$用一个长度位9的二进制数表示当前行、当前列、当前位置所在的小九宫格的状态。\\\qquad'0'表示当前位的数字不能填，'1'表示可填。\\\qquad因此开长度为9的数组row和col来存储每一行的状态，3×3的方阵cell来存储九个小九宫格的状态。$

$②、数组ones来记录每种状态中'1'的个数，也就是可供选择的数字的个数。\\\qquad同时要预处理数组快速查询log_2X，来查看具体是哪一位上的'1'，也就是填了哪个数字。$

$③、draw函数来进行在位置(x,y)上填写数字和删去已填写的数字的操作。\\\qquad填写数字就是在对应的行、列、九宫格的状态所对应的位置填'0'。删去这个数就是在该位填'1'$

$④ 、 g e t 函数用来求位置 (x, y) 能够填的数的状态。等于行、列、九宫格状态相与。$

$⑤、每次操作我们先查找所有空位中，可供选择的数字的个数最少的空位的位置。\\\qquad计算这个位置的状态，接着枚举这个位置能填的数字，开始dfs。$

代码：

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;

const int N=9,M=1<<9;

int ones[M],log_2[M];
int row[N],col[N],cell[3][3];
char str[10][10];

void cal()
{
    for(int i=0;i<M;i++)
        for(int j=0;j<N;j++)
            ones[i]+=(i>>j&1);
    for(int i=0;i<N;i++) log_2[1<<i]=i;
}

void Init()
{
    for(int i=0;i<N;i++)  row[i]=col[i]=(1<<N)-1;
    for(int i=0;i<3;i++)
        for(int j=0;j<3;j++)
            cell[i][j]=(1<<N)-1;
}

void draw(int x,int y,int t,bool is_set)
{
    if(is_set) str[x][y]='1'+t;
    else str[x][y]='0';
    
    int v=1<<t;
    if(!is_set) v=-v;  //删除第t位
    
    row[x]-=v;
    col[y]-=v;
    cell[x/3][y/3]-=v;
}

int lowbit(int x)
{
    return x & -x;
}

int get(int x,int y)  //计算(x,y)能填哪些数
{
    return row[x] & col[y] & cell[x/3][y/3];
}

bool dfs(int cnt)
{
    if(!cnt) return true;
    
    int minv=10;  //计算能填的数的个数最少的位置
    int x,y;  
    for(int i=0;i<N;i++)
        for(int j=0;j<N;j++)
            if(str[i][j]=='0')
            {
                int st=get(i,j);
                if(ones[st]<minv)
                {
                    minv=ones[st];
                    x=i,y=j;
                }
            }
    
    int st=get(x,y);
    for(int i=st;i;i-=lowbit(i))
    {
        int t=log_2[lowbit(i)];
        draw(x,y,t,true);
        if(dfs(cnt-1)) return true;
        draw(x,y,t,false);   //回溯
    }
    
    return false;
}

int main()
{
    cal();
    
    int T;
    cin>>T;
    while(T--)
    {
        Init();
        for(int i=0;i<9;i++) cin>>str[i];
        
        int cnt=0;  //需要填的数的个数
        for(int i=0;i<N;i++)     //(i,j)是二维坐标，k是一维坐标
            for(int j=0;j<N;j++)
                if(str[i][j]!='0')
                {
                    int t=str[i][j]-'1';
                    draw(i,j,t,true);
                }
                else cnt++;
                
        dfs(cnt);
        
        for(int i=0;i<9;i++)
            printf("%s\n",str[i]);
    }
    
    return 0;
}