本来想直接套用八皇后的代码,修改后发现n>=10后num=0,找不到问题点,如果大家能发现什么问题麻烦告诉我:
错误代码如下:
public class Solution {
int num = 0;
public int totalNQueens(int n) {
int[] isused = new int[n]; //每个位置是否被走过,是就是1,否就是0
int[] arr = new int[n];
StringBuffer sb = new StringBuffer();
for (int i = 0; i < n; i++) {
arr[i]=i+1;
}
help(arr, isused, sb);
return num;
}
//结果是否是否符合
public boolean check(StringBuffer sb) {
for (int i = 0; i < sb.length(); i++) {
for (int j = i+1; j < sb.length(); j++) {
if(sb.charAt(i)==sb.charAt(j)||i-j==sb.charAt(i)-sb.charAt(j)||j-i==sb.charAt(i)-sb.charAt(j)) {
//两个棋子在同一个线或斜线上,抛弃此路线
return false;
}
}
}
return true;
}
public void help(int[] arr,int[] isused,StringBuffer sb) {
if(check(sb)==true&&sb.length()==arr.length) {
num++;
return;
}
for (int i = 0; i < arr.length; i++) {
if(isused[arr[i]-1]==1||check(sb)==false) continue;
sb.append(arr[i]);
isused[arr[i]-1]=1;
help(arr, isused, sb);
sb.deleteCharAt(sb.length()-1);
isused[arr[i]-1]=0;
}
}
}
后来参考了其他人n皇后的做法:
public class Solution {
int num = 0;
int[] x ;
public int totalNQueens(int n) {
int[] isused = new int[n]; //每个位置是否被走过,是就是1,否就是0
x = new int[n+1];
help(1, n);
return num;
}
//结果是否是否符合
public boolean check(int i) {
for (int j = 1; j < i; j++) {
if(x[j]==x[i]||Math.abs(i-j)==Math.abs(x[i]-x[j])) {
//两个棋子在同一个线或斜线上,抛弃此路线
return false;
}
}
return true;
}
public void help(int t,int n) {
if(t>n) {
num++;
return;
}
for (int i = 1; i <=n; i++) {
x[t] = i;
if(check(t)) {
help(t+1, n);
}
}
}
}
简洁了很多,但是自己却还是刻板的套用8皇后的问题,所以实际上这个回溯我自己还是没有完全弄懂,只了解了一道题的代码,但不会举一反三,不灵活。