【洛谷】P1784 数独解法

此题解法较简单，用了深度优先搜索的思想，缺点是多层函数调用在空间复杂度上较大，暂时未进行改进，C++代码如下：

#include <iostream>
using namespace std;
/*
sudoku solver
By Felix Chang
*/
bool success = false;
int nums[9][9];
bool check(int key,int pos)
{
    int row = pos / 9;
    int col = pos % 9;
    for(int i=0;i<9;++i)
    {
        if(nums[row][i] == key) return false;
    }
    for(int i=0;i<9;++i)
    {
        if(nums[i][col] == key) return false;
    }
    for(int i=row/3*3;i<row/3*3+3;++i)
    {
        for(int j=col/3*3;j<col/3*3+3;++j)
        {
            if(nums[i][j] == key) return false;
        }
    }
    return true;
}

//n from 0 to 80
void search(int pos)
{
    //for 1-9, try it.If proper or blank,search(n+1);
    int row = pos / 9;
    int col = pos % 9;
    if(pos == 81)
    {
        success = true;
        return;//exit
    }
    if(nums[row][col] != 0)
        search(pos+1);
    else
    {
        for(int i=1;i<10;++i)
        {
            if(check(i,pos) == true)
            {
                nums[row][col] = i;
                search(pos+1);
                if(success)
                    return;//exit2
                nums[row][col] = 0;
            }
        }
    }
}

int main()
{
    for(int i=0;i<9;++i)
    {
        for(int j=0;j<9;++j)
        {
            cin >> nums[i][j];
        }
    }
    search(0);
    for(int i=0;i<9;++i)
    {
        for(int j=0;j<9;++j)
        {
            cout << nums[i][j] << ' ';
            if(j==8) cout << endl;
        }
    }
    return 0;
}