[LeetCode]51 N-Queens

2243 단어 LeetCodeNP
https://oj.leetcode.com/problems/n-queens/
http://blog.csdn.net/linhuanmars/article/details/20667175
public class Solution {
    public List<String[]> solveNQueens(int n) {
        
        List<String[]> result = new ArrayList<>();
        int[] qcols = new int[n];
        solve(n, 0, qcols, result);
        return result;
        
    }
    
    private void solve(int n, // number of queen
                       int row, // current row iteration, [0 -> n-1]
                       int[] qcols, // describing queens positions
                       List<String[]> result)
    {
        if (row == n)
        {
            // all [0, n-1] has fulfilled, a solution found.
            result.add(toBoard(qcols));
            return;
        }
        
        // We are on row
        // try to put Q in each columns, and check
        for (int i = 0 ; i < n ; i ++)
        {
            qcols[row] = i;
            boolean valid = checkBoard(qcols, row);
            if (valid)
            {
                // Enter next iteration
                solve(n, row + 1, qcols, result);
            }
        }
    }
    
    //         Q(row)         
    //          Q   ,       
    //    :qcols[i] != qcols[row]
    //       abs(i - row) != abs(qcols[i] - qcols[row])
    // Check new Q(on row) can fit existing board.
    private boolean checkBoard(int[] qcols, int row)
    {
        for (int i = 0 ; i < row ; i ++)
        {
            if ((qcols[i] == qcols[row]) ||
               Math.abs(i - row) == Math.abs(qcols[i] - qcols[row]))
               return false;
        }
        return true;
    }

    // Build the board.    
    private String[] toBoard(int[] qcols)
    {
        int n = qcols.length;
        String[] b = new String[n];
        for (int i = 0 ; i < n ; i ++)
        {
            int qcol = qcols[i];
            StringBuilder sb = new StringBuilder();
            for (int j = 0 ; j < qcol ; j ++)
                sb.append(".");
            sb.append("Q");
            for (int j = qcol + 1 ; j < n ; j ++)
                sb.append(".");
            b[i] = sb.toString();
        }
        return b;
    }
}

좋은 웹페이지 즐겨찾기