52. N-Queens II
一、题目
1、审题
2、分析:
输入数字n, 求 n-皇后问题存在几个解。
二、解答
1、思路:
同上一题,只是采用一个参数对解的个数进行记录,最终 DFS 返回的即为所求解个数。
class Solution { public int totalNQueens(int n) { char[][] arr = new char[n][n]; for (int i = 0; i < n; i++) { for (int j = 0; j < n; j++) { arr[i][j] = '.'; } } int total = helper(arr, n, 0, 0); return total; } private int helper(char[][] arr, int n, int row, int total) { if(row == n) { return ++total; } else { for (int column = 0; column < n; column++) { if(isValid2(arr, row, column, n)) { arr[row][column] = 'Q'; total = helper(arr, n, row+1, total); arr[row][column] = '.'; } } } return total; } private boolean isValid2(char[][] arr, int row, int column, int n) { // check the column for (int i = 0; i < row; i++) if(arr[i][column] == 'Q') return false; // check the 45 o for (int i = row -1, j = column - 1; i >= 0 && j >= 0; i--, j--) if(arr[i][j] == 'Q') return false; // check the 135 o for(int i = row -1, j = column + 1; i >= 0 && j < n; --i, ++j) if(arr[i][j]== 'Q' ) return false; return true; } }