解数独java实现

leetcode37 力扣37题解法 力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

这题属于N皇后的进阶版，主要也是用boolean数组判断冲突状态和回溯，同学们可以先看我上一篇的N皇后解法:N皇后java实现_追求欲望而非欲望的对象的博客-CSDN博客

首先介绍一个概念 “回溯”：通过某些手段回到程序运行的上一个阶段。

举例：以下代码中，程序在递归调用前后状态保持不变，就是回溯。

 

  public static void main(String[] args) {
        rec(1, new LinkedList<>());
    }
 
    static void rec(int n, LinkedList<String> list) {
        if (n == 3) {
            return;
        }
        System.out.println("before:" + list);
        list.push("a");
        rec(n + 1, list);
        list.pop();
        System.out.println("after:" + list);
    }

先上代码，再详细解读：

    static void solveSudoku(char[][] table) {
        boolean[][] row = new boolean[9][9]; //行冲突
        boolean[][] column = new boolean[9][9]; //列冲突
        boolean[][] square = new boolean[9][9];//3x3宫格冲突
        for (int i = 0; i < 9; i++) {
            for (int j = 0; j < 9; j++) {
                char c = table[i][j];
                if (c != '.') {
                    row[i][c - '1'] = column[j][c - '1'] = square[i / 3 * 3 + j / 3][c - '1'] = true;
                }
            }
        }
        dfs(0, 0, table, row, column, square);
    }

    static boolean dfs(int i, int j, char[][] table, boolean[][] row, boolean[][] column, boolean[][] square) {
        while (table[i][j] != '.') {
            if (++j > 8) {
                j = 0;
                i++;
            }

            if (i>=9) {
                return true;
            }
        }
        for (int k = 1; k <= 9; k++) {
            if (row[i][k - 1] || column[j][k - 1] || square[i / 3 * 3 + j / 3][k - 1]) {//冲突校验
                continue;
            }
           table[i][j] = (char) (k + '0');
            row[i][k - 1] = column[j][k - 1] = square[i / 3 * 3 + j / 3][k - 1] = true;
            if (dfs(i, j, table, row, column, square)) {
                return true;
            }
            row[i][k - 1] = column[j][k - 1] = square[i / 3 * 3 + j / 3][k - 1] = false;
            table[i][j] = '.';//回溯
        }
        return false;
    }

    public static void main(String[] args) {
        char[][] table = {
                {'5', '3', '.', '.', '7', '.', '.', '.', '.'},
                {'6', '.', '.', '1', '9', '5', '.', '.', '.'},
                {'.', '9', '8', '.', '.', '.', '.', '6', '.'},
                {'8', '.', '.', '.', '6', '.', '.', '.', '3'},
                {'4', '.', '.', '8', '.', '3', '.', '.', '1'},
                {'7', '.', '.', '.', '2', '.', '.', '.', '6'},
                {'.', '6', '.', '.', '.', '.', '2', '8', '.'},
                {'.', '.', '.', '4', '1', '9', '.', '.', '5'},
                {'.', '.', '.', '.', '8', '.', '.', '7', '9'}
        };
        solveSudoku(table);

        print(table);
    }

    static char[][] solved = {
            {'5', '3', '4', '6', '7', '8', '9', '1', '2'},
            {'6', '7', '2', '1', '9', '5', '3', '4', '8'},
            {'1', '9', '8', '3', '4', '2', '5', '6', '7'},
            {'8', '5', '9', '7', '6', '1', '4', '2', '3'},
            {'4', '2', '6', '8', '5', '3', '7', '9', '1'},
            {'7', '1', '3', '9', '2', '4', '8', '5', '6'},
            {'9', '6', '1', '5', '3', '7', '2', '8', '4'},
            {'2', '8', '7', '4', '1', '9', '6', '3', '5'},
            {'3', '4', '5', '2', '8', '6', '1', '7', '9'}
    };

    static void print(char[][] table) {
        for (char[] chars : table) {
            System.out.println(new String(chars));
        }
        System.out.println(Arrays.deepEquals(table, solved));
    }

代码中所有i表示行，所有j表示列

三个boolean[][]数组的第一维代表九宫格的第几行（row），第几列(column)，和第几个3x3方格(square)，索引0代表数字1,以此类推。第二维代表具体哪个数字是被占用（冲突）状态。

行和列的的索引比较好找，九宫格的索引与i j 的关系符合公式:索引= i/3*3+j/3;

首先遍历每个数字，并对冲突情况进行初始化。

对于0-9的字符，char类型底层的int值也是连续排列的。c-'1'可以得到正确索引。

  for (int i = 0; i < 9; i++) {
            for (int j = 0; j < 9; j++) {
                char c = table[i][j];
                if (c != '.') {
                    row[i][c - '1'] = column[j][c - '1'] = square[i / 3 * 3 + j / 3][c - '1'] = true;
                }
            }
        }

其次 在每次递归中寻找下一个'.',用来填充数字，当i=9时即意味着已经遍历所有格子并且没有回溯。

 while (table[i][j] != '.') {
            if (++j > 8) {
                j = 0;
                i++;
            }

            if (i>=9) {
                return true;
            }
        }

在每个格子里尝试所有的数字，发现没有冲突的数字则进行递归，一旦在某次方法调用的循环里全部执行了continue，则此路不通，将所有格子和冲突回溯到上一次方法调用的状态，最后能成功执行到最后一个格子，表示在此之前，每个格子至少有一个解，最后数组就是最终解。

   for (int k = 1; k <= 9; k++) {
            if (row[i][k - 1] || column[j][k - 1] || square[i / 3 * 3 + j / 3][k - 1]) {
                continue;
            }
           table[i][j] = (char) (k + '0');
            row[i][k - 1] = column[j][k - 1] = square[i / 3 * 3 + j / 3][k - 1] = true;
            if (dfs(i, j, table, row, column, square)) {
                return true;
            }
            row[i][k - 1] = column[j][k - 1] = square[i / 3 * 3 + j / 3][k - 1] = false;
            table[i][j] = '.';
        }
        return false;

比如程序到了绿色方块位置，把1-9都试过了还是不行，只能回溯到上一层方法，找到上一层的另一个解，在到下一层去遍历1-9。