LeetCode-51. N-Queens(Hard):
The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.
Given an integer n, return the number of distinct solutions to the n-queens puzzle.
Example:
Input: 4 Output: 2 Explanation: There are two distinct solutions to the 4-queens puzzle as shown below. [ [".Q..", // Solution 1 "...Q", "Q...", "..Q."], ["..Q.", // Solution 2 "Q...", "...Q", ".Q.."] ]
我的解法:
52题是51题的简化,答案不用输出八皇后问题的解的排列,只需要输出解的个数。
由于51题是看了答案,52题决定自己做一下。不做不知道,一做就卧槽。对这个问题的解法有了更深刻的理解,其中51题答案没看懂的部分也看懂了。
一共两个函数,totalNQueens中首先定义三个数组,分别记录列、左上到右下的斜排(2n-1排)、右上到左下的斜排(2n-1排)的占用情况。然后调用NQueens返回答案。NQueens中按行依次遍历,循环的方式(依次尝试)确定第一排皇后的位置,记录列和斜排的占用情况,当然行就不用记录了,因为是一行一行寻找皇后可能的安放位置,行是不冲突的。然后递归调用NQueens确定下一行皇后的位置(row+1),如果row == n,就代表之前从0到n-1的n行都找到了皇后的位置,那么就找到了一种可行的排列,ans++。因为是回溯问题,所以每一个递归结束后需要恢复状态(代码中for循环最后三行)。回溯问题的关键就在于状态的记录和恢复。
class Solution {
public int totalNQueens(int n) {
boolean[] col = new boolean[n];
boolean[] topleftTobottomright = new boolean[2*n-1];
boolean[] toprightTobottomleft = new boolean[2*n-1];
int ans = 0;
ans = NQueens(ans, n, 0, col, topleftTobottomright, toprightTobottomleft);
return ans;
}
public int NQueens(int ans, int n, int row, boolean[] col, boolean[] tlTobr, boolean[] trTobl) {
if(row == n) {
ans = ans + 1;
//System.out.println(ans);
return ans;
}
for(int i = 0; i < n; i++) {
if(col[i] || tlTobr[i+row] || trTobl[n-1 + row-i]) {
continue;
}
col[i] = true;
tlTobr[i+row] = true;
trTobl[n-1 + row-i] = true;
ans = NQueens(ans, n, row+1, col, tlTobr, trTobl);
col[i] = false;
tlTobr[i+row] = false;
trTobl[n-1 + row-i] = false;
}
return ans;
}
}