leetcode 37. 解数独

@(labuladong的算法小抄)[回溯]

leetcode 37. 解数独

题目描述

解题思路

参考：labuladong的算法小抄P301

解数独和n皇后思路的区别：

n皇后需要遍历的路径是所有行，选择列表是某一行的每一列，因为n皇后是要将Q放在这一行的某一个列上；

而解数独需要遍历的路径是9*9的每一个格子，选择列表是1~9这九个数字，因为解数独是要填满每个格子，每个格子有9种选择。

class Solution {
    public void solveSudoku(char[][] board) {
        backtrack(board, 0, 0);
    }
    /* 由于仅有唯一解，找到唯一解后直接返回，所以不需要全部遍历 */
    private boolean backtrack(char[][] board, int row, int col) {
        /* base case：找到一个可行解 */
        if (row == board.length) {
            return true;
        }
        /* 如果列到头了，则到下一行重新开始遍历 */
        if (col == board.length) {
            return backtrack(board, row + 1, 0);
        }

        /* 当前位置上的所有选择就是1~9这些数字 */
        for (char ch = '1'; ch <= '9'; ch++) {
            /* 如果原本就是数字，则直接跳过这一列 */
            if (board[row][col] != '.') {
                return backtrack(board, row, col + 1);
            }
            /* 排除不合法的选择 */
            if (!isValid(board, row, col, ch)) continue;

            board[row][col] = ch;
            /* 做出当前选择后，继续去下一个位置，如果找到唯一解，直接返回 */
            if (backtrack(board, row, col + 1)) {
                return true;
            }
            board[row][col] = '.';
        }

        return false;
    }
    /* 判断board[row][col]是否可以放入数字n */
    private boolean isValid(char[][] board, int row, int col, char n) {
        for (int i = 0; i < board.length; i++) {
            /* 判断该行是否存在重复 */
            if (board[row][i] == n) return false;
            /* 判断该列是否存在重复 */
            if (board[i][col] == n) return false;
            /* 判断3*3方框里是否存在重复 */
            /* board[(row / 3) * 3][(col / 3) * 3]是n所在的3*3方框最上角元素的位置 */
            if (board[(row / 3) * 3 + i / 3][(col / 3) * 3 + i % 3] == n) return false;
        }
        return true;
    }
}