【无标题】

利用回溯法解数独游戏

package com.backtrack;

import static java.lang.String.format;

/**
 * <p>
 * 利用回溯的算法解数独游戏
 */
public class Sudoku75 extends BackTrack {

    public static final int[][] board = {
            { 5, 3, 0, 0, 7, 0, 0, 0, 0 },
            { 6, 0, 0, 1, 9, 5, 0, 0, 0 },
            { 0, 9, 8, 0, 0, 0, 0, 6, 0 },
            { 8, 0, 0, 0, 6, 0, 0, 0, 3 },
            { 4, 0, 0, 8, 0, 3, 0, 0, 1 },
            { 7, 0, 0, 0, 2, 0, 0, 0, 6 },
            { 0, 6, 0, 0, 0, 0, 2, 8, 0 },
            { 0, 0, 0, 4, 1, 9, 0, 0, 5 },
            { 0, 0, 0, 0, 8, 0, 0, 7, 9 },
    };

    // (i,j)坐标中的九宫格所在的开始位置,如何(4,7)的值就是[3,6]
    public static final int[][][] sudokuMap = new int[9][9][2];

    /**
     * <p>
     * 利用回溯法解数独游戏
     * 
     * @param board
     * @return
     */
    public static boolean backtraking(int[][] board) {
        for (int i = 0; i < 9; i++) {
            for (int j = 0; j < 9; j++) {
                // 格子中的数字>0的话，那么跳过继续下一个格子尝试放置数字
                if (board[i][j] > 0) {
                    continue;
                }
                // 尝试在位置(i, j)坐标 放9个数字
                for (int num = 1; num <= 9; num++) {
                    if (isValid(i, j, num)) {
                        board[i][j] = num;
                        boolean success = backtraking(board);
                        if (success) {
                            return true;
                        }
                        // 回溯
                        board[i][j] = 0;
                    }
                }
                // 如果当前格子位置9个数字都尝试放置，都没有成功，那么当前棋盘不合格，有问题
                return false;
            }
        }
        return true;
    }

    /**
     * <p>
     * 初始化每个九宫格的位置的开始的位置
     */
    public static void initSudokuMap() {
        for (int i = 0; i < sudokuMap.length; i++) {
            for (int j = 0; j < sudokuMap.length; j++) {
                // 每个九宫格开始的横向和竖向位置
                int rowStartIndex = i / 3 * 3;
                int columnStartIndex = j / 3 * 3;

                sudokuMap[i][j] = new int[] { rowStartIndex, columnStartIndex };
            }
        }
    }

    /**
     * <p>
     * 在(i, j)位置放置数字num是否合法
     * <p>
     * 横向和竖向都不能有重复的数字，并且当前的九宫格内也不能有重复的数字
     * 
     * @param args
     */
    public static boolean isValid(int row, int column, int num) {
        // 如果board位置本身已经有了>0的数字，那么直接不合法
        if (board[row][column] > 0) {
            return false;
        }
        // i行中是否有重复元素
        for (int j = 0; j < 9; j++) {
            if (board[row][j] == num) {
                return false;
            }
        }
        // j列中是否有重复的元素
        for (int i = 0; i < 9; i++) {
            if (board[i][column] == num) {
                return false;
            }
        }
        // 当前所在的九宫格内是否有重复元素。
        int sudokuRowStart = sudokuMap[row][column][0];
        int sudokuColumnStart = sudokuMap[row][column][1];

        // 判断九宫格中是否有重复元素
        // i行中是否有重复元素
        for (int i = sudokuRowStart; i < sudokuRowStart + 3; i++) {
            // j列中是否有重复的元素
            for (int j = sudokuColumnStart; j < sudokuColumnStart + 3; j++) {
                if (board[i][j] == num) {
                    return false;
                }
            }
        }
        return true;
    }

    /**
     * 打印棋盘
     */
    public static void printBoard() {
        for (int i = 0; i < 9; i++) {
            for (int j = 0; j < 9; j++) {
                System.out.print(format("%s  ", board[i][j]));
            }
            System.out.println();
        }
    }

    public static void main(String[] args) {
        initSudokuMap();
        int row = 3;
        int column = 5;
        for (int i = 0; i < sudokuMap[row][column].length; i++) {
            System.out.print(format("%s\t", sudokuMap[row][column][i]));
        }

        System.out.println("=============================");
        System.out.println(isValid(6, 7, 1));

        boolean success = backtraking(board);
        if (success) {
            printBoard();
        }
    }

}