leetcode-Sudoku Solver

最新推荐文章于 2021-03-16 11:12:31 发布

原创最新推荐文章于 2021-03-16 11:12:31 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #面试 #启发式搜索 #数独

acm-图论同时被 3 个专栏收录

27 篇文章

订阅专栏

面试

22 篇文章

订阅专栏

leetcode

22 篇文章

订阅专栏

本文介绍了一种基于启发式搜索的数独求解算法。通过记录每个空格可能的数值种类，并按照可能性大小进行排序遍历，实现了对数独的有效求解。文中详细解释了如何利用三个集合记录行、列及3x3宫格内的数字信息，从而快速确定空格的候选数值。

Write a program to solve a Sudoku puzzle by filling the empty cells.

Empty cells are indicated by the character '.'.

You may assume that there will be only one unique solution.

A sudoku puzzle...

...and its solution numbers marked in red.

题意：给出一个数独，求解，保证有唯一解

分析：简单的思路是，遍历每个空格，枚举所有可能取值，递归求解。为了减小搜索空间，用启发式搜索。即，记录每个空格可能取值的种类，按从小到大遍历所有空格，保证搜索空间最小。用三个set[9]分别记录每行，每列，每个3*3区域已有的数字，然后就能在O(1)内对相应的set做并集，求出任一空格可能的取值。

代码：

int cmp(const void *a, const void *b)
{
    return ((int*)a)[2]-((int*)b)[2];
}

class Solution {
private:
    set<char> row[9],col[9],batch[9];
    int op_qu[90][3],qu_cnt;


public:
    bool fill(vector<vector<char> > &bd,int num)
    {
        if(num>=qu_cnt) return true;
        int i=op_qu[num][0],j=op_qu[num][1];
        set<char> tmp,now;
        set_union(row[i].begin(),row[i].end(),col[j].begin(),col[j].end(),inserter(tmp,tmp.begin()));
        int k = i/3*3+j/3;
        set_union(tmp.begin(),tmp.end(),batch[k].begin(),batch[k].end(),inserter(now,now.begin()));

        for(char ch='1'; ch<='9'; ch++)
        {
            if(now.find(ch)==now.end())
            {
                bd[i][j] = ch;
                row[i].insert(ch);
                col[j].insert(ch);
                batch[k].insert(ch);
                if(fill(bd,num+1))
                    return true;
                row[i].erase(ch);
                col[j].erase(ch);
                batch[k].erase(ch);
            }
        }
        return false;
    }
    void solveSudoku(vector<vector<char> > &board) {

        for(int i=0; i<9; i++)
        {
            row[i].clear();
            col[i].clear();
            batch[i].clear();
        }
        qu_cnt = 0;

        for(int i=0; i<9; i++)
        {
            for(int j=0; j<9; j++)
            {
                char ch = board[i][j];
                if(ch!='.')
                {
                    row[i].insert(ch);
                    col[j].insert(ch);
                    batch[i/3*3+j/3].insert(ch);
                }
            }
        }
        for(int i=0; i<9; i++)
        {
            for(int j=0; j<9; j++)
            {
                char ch = board[i][j];
                if(ch=='.')
                {
                    set<char> tmp,now;
                    set_union(row[i].begin(),row[i].end(),col[j].begin(),col[j].end(),inserter(tmp,tmp.begin()));
                    int k = i/3*3+j/3;
                    set_union(tmp.begin(),tmp.end(),batch[k].begin(),batch[k].end(),inserter(now,now.begin()));

                    op_qu[qu_cnt][0] = i;
                    op_qu[qu_cnt][1] = j;
                    op_qu[qu_cnt][2] = 9-now.size();
                    qu_cnt++;
                }
            }
        }
        qsort(op_qu,qu_cnt,sizeof(int)*3,cmp);

        fill(board,0);
    }
};