和第51题一样,只不过返回值改为了解的数量。
那么就直接套用第51题的代码好了
本来想直接用Integer的,但是Integer对象自增后,并不会改变自身这个对象的值,它的原理是创建一个新的Integer对象,并用新的地址覆盖掉旧的地址,而方法传参我们需要的是同一个指针,因此不用Integer
代码:
public int totalNQueens(int n) {
List<Integer> forEver = new ArrayList<>();
List<Integer> next = new ArrayList<>();
//不能使用Integer,因为Integer对象的值想进行改变,就一定会返回新的Integer对象
//比如res++执行完成后,它的地址改变了,不再是原来的那个res了
//Integer res=0;
int[] res = new int[1];
dfs(res, forEver, next, n);
return res[0];
}
public void dfs(int[] res, List<Integer> forEver, List<Integer> next, int n) {
for (int i = 0; i < n; i++) {
if (forEver.contains(i) || next.contains(i))
continue;
forEver.add(i);
List<Integer> newNext = new ArrayList<>();
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(res, forEver, newNext, n);
forEver.remove(forEver.size() - 1);
}
if (forEver.size() == n) {
res[0]=res[0]+1;
}
}