nQueen问题java实现

最近一个项目刚刚结束，稍稍清闲了下来。其实我对自己的编程技术一直不太自信，作为程序员，在大学期间竟然都没选修过算法设计，真可谓三脚猫功夫。

正因如此，我反而一直保留着一本算法设计的书（数据结构，高数啥的大学毕业的那年遗忘在宿舍了 - -！），近来随手翻翻，找了个问题（nQueen）自己练手，用java实现之，请各位轻拍。

问题简单描述下：在n*n的棋盘上放置n个棋子（Queen）使得同一列，行，斜线上有且仅有一个Queen。更多该问题的背景可以百度之。

/**
 * 请调用静态方法NQueen.calc(int n)方法计算n皇后问题
 * @author Lovememo
 *  
 */
public class NQueen {
    private int[][] chessBoard;
    private int succeedSum;      //记录解的个数
    
    private void setQueen(int m, int n, boolean flag) {
        if(flag == true) 
            this.chessBoard[m][n] = 1;
         else
            this.chessBoard[m][n] = 0;
    }
    
    private boolean isSolution() {
        int sum = 0;
        for(int i=0; i<this.chessBoard.length; i++) {
            for(int j=0; j<this.chessBoard.length; j++) {
                sum += this.chessBoard[i][j];
            }
        }
        if(sum == this.chessBoard.length)
            return true;
        else
            return false;
    }
    
    private boolean canBeQueen(int m, int n) {
        for(int i=0; i<this.chessBoard.length; i++) {
            for(int j=0; j<this.chessBoard.length; j++) {
                if(1 == this.chessBoard[i][j]) {
                    //横坐标相同，纵坐标相同，或者在同一斜线上，则不能为Queen
                    if(m==i || n==j || Math.abs(m-i) == Math.abs(n-j)) 
                        return false;
                }
            }
        }
        return true;
    }
    
    //递归回溯计算
    private void calculate(int startNum) {
        int chessBoardLength = this.chessBoard.length;
        if(startNum == chessBoardLength-1) { //设置跳出递归条件
            for(int i=0; i<chessBoardLength; i++) {
                if(this.canBeQueen(startNum, i)) {
                    this.setQueen(startNum, i, true);
                    if(this.isSolution()) {
                        this.succeedSum += 1;
                        this.printChess();                        
                    }
                    this.setQueen(startNum, i, false);
                    return;
                }
            }
            return;
        }
        
        for(int i=0; i<chessBoardLength; i++) {
            if(this.canBeQueen(startNum, i)) {
                this.setQueen(startNum, i, true);
                this.calculate(startNum + 1);
                this.setQueen(startNum, i, false);
            }
        }        
    }    
    
    //将n皇后问题的一个解打印出来
    private void printChess() {
        for(int i=0; i<this.chessBoard.length; i++) {
            for(int j=0; j<this.chessBoard.length; j++) {
                System.out.print(this.chessBoard[i][j] + " ");
            }
            System.out.println();
        }
        System.out.println();
    }

    private NQueen(int n) {        
        this.chessBoard = new int[n][n];
        this.succeedSum = 0;
    }
    
    public static void calc(int n) {
        NQueen nq = new NQueen(n);
        nq.calculate(0);
        System.out.println(nq.succeedSum);
    }

    public static void main(String[] args) {
        NQueen.calc(8);
        
    }
}