首页 > 代码库 > LeetCode----Sudoku Solver+精确覆盖问题解法(Dancing Links)

LeetCode----Sudoku Solver+精确覆盖问题解法(Dancing Links)

BackGround:

    做完LeetCode上的数独题目好长时间了,今天将做题时参考的Algorithm X 以及 Dancing Links 整理出来。话说理解算法+写出程序一共用了三天,智商果然余额不足。。。


介绍:

    由于Dancing Links 是为了解决数独问题学习的,那就从数独问题下手,围绕数独问题展开对Algorithm X 和 Dancing Links的介绍,最后将数独问题的解法以及Java源码放出来。


精确覆盖问题:

    没错,解数独的问题就是一个解  精确覆盖   问题的过程,即数独问题是精确覆盖问题的一个特例。

    精确覆盖问题可以用下面的例子来说明:

    假设有集合S = { a , b , c , d , e , f } , 还有一些S的子集 S1 = { a , c , f} , S2 = {b , d } , S3 = { e } , S4 = { b , d , f , e} ,从这些子集中选出n个集合,使得这些集合的并集为S , 并且子集中的元素有且只能出现一次。很容易看出我们可以选择S1 、 S2 、 S3 。 但是不能选择 S1、 S4 因为f元素重复出现了。


精确覆盖问题的形式化表达:

    我们可以用一个01矩阵来形式化表达精确覆盖问题,还是上面的例子,我们可以表达为如下的矩阵:


每一行代表一个子集,那么问题就是选择某些行,使得a-f列出现且仅出现一个1。


精确覆盖问题的解法:

    Knuth大神给出的解法就是Algorithm X 。 其思想就是从01矩阵入手。解法的过程(不考虑回溯的次数多少)为:

设01矩阵为A


如果 A 是空的,


             问题解决;成功终止。


否则,


             选择一个列 c(确定的)
             选择一个行 r,满足 A[r, c]=1 (不确定的)
             把 r 包含进部分解
             对于所有满足 A[r,j]=1 的 j
                          从矩阵 A 中删除第 j 列;
                          对于所有满足 A[i,j]=1 的 i,
                          从矩阵 A 中删除第 i 行。


在不断减少的矩阵 A 上递归地重复上述算法。


这个算法的有点麻烦,其实过程很简单。用上面的例子来解释就是


1、选择一个列,这里我们选择a列,找到这一列中有1的所有行,记为row1 , row2 , row3....

2、首先选择row1作为解的一部分,这里为选择第一行,然后将这一行中所有  为1的  列删除(这个行中已经满足了这些列有1)。在上面的例子中就是a , c , f列。

3、在删除a 、 c 、 f 列的过程中,还要删除这一列中有1的行(避免当前列中重复出现1)。以删除f列为例,由于选择第一行了,所以f列的1不能重复出现,即最后一行不能再选择了,所以要删除最后一行。

4、重复上面的1-3步骤,如果矩阵最后为空,那么问题已经解决。否则,说明row1不能选择,我们再回溯问题,不选择row1,选择row2。


X算法的实现:

算法有了,那么怎么实现呢?我们可以递归解决问题,每一步复制一个矩阵,但是如果搜索次数很多的话,需要的空间会比较大。所以又有了Dancing links 的实现(后面叫DLX)。DLX虽然听起来很麻烦,但是他实现起来确实麻烦-,-!本质上DLX就是一个双向十字链表。如下(图是挖别人的。。):


h是整个链表的表头,A、B、.....G是每一列的列头,其他的是节点。用这个链表来代表上述的01矩阵(这一步应该很容易理解)。然后将矩阵的操作转化为链表的操作。


用DLX的好处是:

1、X算法在链表的列头上进行搜索。==========》如果需要删除某一列,直接删除对应的列头即可。

2、可以方便的进行恢复,假如要删除C列,只需要两个操作便可删除   

 C.L.R = C.R       

 C.R.L = C.L

恢复时

  C.R.L = C.L

  C.L.R = C.R

3、扫描列是否都满足时,不必扫描全部的列,只需判断表头.L 是否 表头即可。


代码:

用Java写的LeetCode的解题数独的解题代码。

public class Solution {
    private static final int colmns = 324; //9 * 9 * 3 + 81
    private Node head;
    private Node[] cols;
    private int[] solution = new int[81];
    private boolean flag =false;

    String sudoku = "53..7...." +
            "6..195..." +
            ".98....6." +
            "8...6...3" +
            "4..8.3..1" +
            "7...2...6" +
            ".6....28." +
            "...419..5" +
            "....8..79";

    char[][] sudoku1;
    private int[][] sudokuArray = new int[9][9];

    ////////////////////////THE LEETCODE MAIN//////////////////////
    public void solveSudoku(char[][] board) {
        StringBuilder sb = new StringBuilder();
        for(char[] t : board){
            for(char tt : t) {
                sb.append(tt);
            }
        }
        sudoku1 = board;
        init();
        construct(sb.toString());
        search(0);

    }
    ///////////////////////////////////////////////////////////////
    public void init(){
        head = new Node();
        head.L = head;
        head.R = head;
        cols = new Node[colmns];
        for(int i = 0 ; i < cols.length ; i++){
            Node temp = new Node();
            temp.C = temp;
            temp.U = temp;
            temp.D = temp;
            temp.R = head;
            temp.L = head.L;
            temp.r = i;
            head.L.R = temp;
            head.L = temp;
            cols[i] = temp;
        }
    }

    public void link(int r , int[] cs){
        Node rowHead = null;

        for(int i = 0 ; i < cs.length ; i++){
            int c = cs[i];
            Node columnHeader = cols[c].C;
            Node temp = new Node();
            temp.r = r;
            if(i == 0){
                rowHead = temp;
                rowHead.R = rowHead;
                rowHead.L = rowHead;
            }
            ///////////
            temp.R = rowHead;
            temp.L = rowHead.L;
            rowHead.L.R = temp;
            rowHead.L = temp;

            ///////////
            temp.D = columnHeader;
            temp.U = columnHeader.U;

            //set the columnHeader
            temp.C = columnHeader;
            columnHeader.U.D = temp;
            columnHeader.U = temp;
            columnHeader.count++;

        }
    }
    /**
     * remove the column c
     * @param c  the column obj
     */
    public void remove(Node c){

        //The column head
        Node cHead = c.C;
        cHead.R.L = cHead.L;
        cHead.L.R = cHead.R;
        //System.out.println("remove -----" + cHead.r);
        //delete the row of this column
        for(Node row = cHead.D ; row != cHead ; row = row.D){
            //delete the row
            for(Node rowNode = row.R ; rowNode != row ; rowNode = rowNode.R){
                rowNode.D.U = rowNode.U;
                rowNode.U.D = rowNode.D;
                rowNode.C.count--;
            }
        }

    }

    /**
     * resume the column
     * @param c
     */
    public void resume(Node c){
        Node cHead = c.C;

        //resume the row
        for(Node row = cHead.U ; row != cHead ; row = row.U){
            for(Node rowNode = row.L ; rowNode != row ; rowNode = rowNode.L){
                rowNode.D.U = rowNode;
                rowNode.U.D = rowNode;
                rowNode.C.count++;
            }
        }

        //resume the head
        cHead.R.L = cHead;
        cHead.L.R = cHead;
    }

    /**
     *
     * @return the column header
     */
    public Node chooseColumn(){
        int min = Integer.MAX_VALUE;
        Node result = null;
        for(Node c = head.R ; c != head ; c = c.R){
            if(c.count < min){
                result = c;
                min = c.count;
            }
        }
        return result;
    }

    public boolean search(int k){
        //System.out.println(java.util.Arrays.toString(solution));
        if(flag == true)
            return true;
        if(head.R == head){
            //System.out.println(java.util.Arrays.toString(solution));
            printSolution1();
            flag = true;
            return true;
        }
        Node c = chooseColumn();
        remove(c);  //remove the c

        for(Node solutionRow = c.D ; solutionRow != c ; solutionRow = solutionRow.D){
            solution[k] = solutionRow.r;  //add the solution

            //remove the column in the right
            for(Node rightNode = solutionRow.R ; rightNode != solutionRow ; rightNode = rightNode.R){
                remove(rightNode);
            }

            //continue searching
            search(k+1);

            //after searching , resume the state
            for(Node leftNode = solutionRow.L ; leftNode != solutionRow ; leftNode = leftNode.L){
                resume(leftNode);
            }


        }

        resume(c);

        return false;
    }

    /**
     * Construct the two direction circular linked list according to the puzzle which is in the form of
     * 2.2.2.2......
     * @param puzzle
     */
    public void construct(String puzzle){
        if(puzzle.length() != 81){
            return;
        }
        for(int i = 0 ; i < puzzle.length() ; i++){
            char current = puzzle.charAt(i);
            int r = i / 9 ;
            int c = i%9;

            if(current == '.'){
                for(int val = 1 ; val <= 9 ; val++){
                    link(r*100 + c * 10 + val , getColumnIndex(r,c,val));
                }
            }else{
                int number = current - '0';
                link(r*100 + c * 10 + number , getColumnIndex(r,c,number));
            }
        }
    }


    public void construct(char[][] puzzle){
        for(int pr = 0 ; pr < puzzle.length ; pr++){
            for(int pc = 0 ; pc < puzzle[pr].length ; pc++){
                int i = pr * 9 + pc;
                char current = puzzle[pr][pc];
                int r = i / 9 ;
                int c = i%9;

                if(current == '.'){
                    for(int val = 1 ; val <= 9 ; val++){
                        link(r*100 + c * 10 + val , getColumnIndex(r,c,val));
                    }
                }else{
                    int number = current - '0';
                    link(r*100 + c * 10 + number , getColumnIndex(r,c,number));
                }
            }
        }
    }

    /**
     * given the row and column and the value , return the four column index of it
     * @param r
     * @param c
     * @param val
     * @return System.out.println(java.util.Arrays.toString(solution));
     */
    public int[] getColumnIndex(int r , int c , int val){
        int[] array = new int[4];
        array[0] = r * 9 + val - 1;
        array[1] = 81 + c * 9 + val - 1;
//        int b = (r / 3) * 3 + c % 3;
        int tr = r / 3;
        int tc = c / 3;
        int b = tr * 3 + tc;
        array[2] = 162 + b * 9 + val - 1;
        array[3] = 243 + r * 9 + c;
        return array;
    }

    public void printSolution(){
        for(int i = 0 ; i < solution.length ; i++){
            int r = solution[i] / 100;
            int c = solution[i] / 10 % 10;
            int val = solution[i] % 10;
            sudokuArray[r][c] = val;
        }
        for(int[] rr : sudokuArray){
            for(int vv : rr){
                System.out.print(vv + "   ");

            }
            System.out.println();
        }
    }
    public void printSolution1(){
        for(int i = 0 ; i < solution.length ; i++){
            int r = solution[i] / 100;
            int c = solution[i] / 10 % 10;
            int val = solution[i] % 10;
            sudoku1[r][c] = (char)(val + '0');
        }
    }
}
class Node{
    int r;   //represent the solution     ****rcv****
    //direction pointer
    Node U;
    Node D;
    Node L;
    Node R;
    //---------
    Node C;  //the header
    int count;  //Used to store the count of 1 in column
}








LeetCode----Sudoku Solver+精确覆盖问题解法(Dancing Links)