题目描述(困难难度)
经典的 N 皇后问题。意思就是摆皇后的位置,每行每列以及对角线只能出现 1 个皇后。输出所有的情况。
解法一 回溯法
比较经典的回溯问题了,我们需要做的就是先在第一行放一个皇后,然后进入回溯,放下一行皇后的位置,一直走下去,如果已经放的皇后的数目等于 n 了,就加到最后的结果中。然后再回到上一行,变化皇后的位置,然后去找其他的解。
期间如果遇到当前行所有的位置都不能放皇后了,就再回到上一行,然后变化皇后的位置。再返回到下一行。
说起来可能还费力些,直接看代码吧。
public List<List<String>> solveNQueens(int n) { List<List<String>> ans = new ArrayList<>(); backtrack(new ArrayList<Integer>(), ans, n); return ans; } private void backtrack(List<Integer> currentQueen, List<List<String>> ans, int n) { // 当前皇后的个数是否等于 n 了,等于的话就加到结果中 if (currentQueen.size() == n) { List<String> temp = new ArrayList<>(); for (int i = 0; i < n; i++) { char[] t = new char[n]; Arrays.fill(t, '.'); t[currentQueen.get(i)] = 'Q'; temp.add(new String(t)); } ans.add(temp); return; } //尝试每一列 for (int col = 0; col < n; col++) { //当前列是否冲突 if (!currentQueen.contains(col)) { //判断对角线是否冲突 if (isDiagonalAttack(currentQueen, col)) { continue; } //将当前列的皇后加入 currentQueen.add(col); //去考虑下一行的情况 backtrack(currentQueen, ans, n); //将当前列的皇后移除,去判断下一列 //进入这一步就是两种情况,下边的行走不通了回到这里或者就是已经拿到了一个解回到这里 currentQueen.remove(currentQueen.size() - 1); } } } private boolean isDiagonalAttack(List<Integer> currentQueen, int i) { // TODO Auto-generated method stub int current_row = currentQueen.size(); int current_col = i; //判断每一行的皇后的情况 for (int row = 0; row < currentQueen.size(); row++) { //左上角的对角线和右上角的对角线,差要么相等,要么互为相反数,直接写成了绝对值 if (Math.abs(current_row - row) == Math.abs(current_col - currentQueen.get(row))) { return true; } } return false; }
时间复杂度:
空间复杂度:
上边我们只判断了列冲突和对角线冲突,至于行冲突,由于我们采取一行一行加皇后,所以一行只会有一个皇后,不会产生冲突。
总
最早接触的一类问题了,学回溯法的话,一般就会以这个为例,所以思路上不会遇到什么困难。