先看一眼题:
n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。
思路:
使用两个数组,一个用来记录永远不能再使用的数字,一个用来存放下一步的时候不能使用的数字。
比如已经使用了1和4
那么下一次放入数字的时候,不能放入1和4,且不能4-1=3,4+1=5 和 1-2=-1,1+2=3
假设n=6,则从0-5中选择不是1,4,-1,3,5的数字,则本次只能放入2
代码:
public List<List<String>> solveNQueens(int n) {
List<Integer> forEver = new ArrayList<>();
List<Integer> next = new ArrayList<>();
List<List<String>> resList = new ArrayList<>();
dfs(resList, forEver, next, n);
return resList;
}
public void dfs(List<List<String>> resList, List<Integer> forEver, List<Integer> next, int n) {
//寻找有没有还能放入的数,有的话则dfs,没有的话看看forEver.size()==n是否成立,成立则创建list加入到resList,否则递归结束
for (int i = 0; i < n; i++) {
if (forEver.contains(i) || next.contains(i))
continue;
forEver.add(i);
List<Integer> newNext = new ArrayList<>();
// newNext.add(i - 1);
// newNext.add(i + 1);
//要加入next中能使两个皇后构成一条斜线的边的顶点,不只是上一个顶点的+1和-1
Integer arr[] = new Integer[forEver.size()];
forEver.toArray(arr);
for (int j = 1, k = arr.length - 1; k > -1; j++, k--) {
newNext.add(arr[k]-j);
newNext.add(arr[k]+j);
}
dfs(resList, forEver, newNext, n);
forEver.remove(forEver.size() - 1);
}
if (forEver.size() == n) {
List<String> list = new ArrayList<>();
Integer[] forEverArr = new Integer[n];
forEver.toArray(forEverArr);
for (int i = 0; i < n; i++) {
char[] chars = new char[n];
//写入Q
chars[forEverArr[i]] = 'Q';
//写入.
for (int j = 0; j < n; j++) {
if (j != forEverArr[i])
chars[j] = '.';
}
//转换为字符串,加入到结果中
String str = new String(chars);
list.add(str);
}
resList.add(list);
}
}
public static void main(String[] args) {
Solution51 s = new Solution51();
s.solveNQueens(4);
}
只打败了13%的人…哎
对于中等和困难难度的题,我还是先以能做出来为目标吧,起码这个题我没看答案自己做出来了。