题目链接:
力扣https://leetcode-cn.com/problems/n-queens-ii/
【分析】回溯法的经典例题,通过一个数组来存储棋子的位置,数组下标为行号,存的值为列号,在改变数组值之前进行判定。
需要注意用java的除法计算斜率的绝对值为1进行判定会出一些小问题。
class Solution {
int ans = 0;
int m;
public boolean is(int x, int y, int a, int b){
if(Math.abs(x - a) == Math.abs(y - b)) return true;
return false;
}
public void backTrack(int t, int arr[]){
if(t == m) {
ans++;
}
for(int i = 0; i < m; i++){
int flag = 1;
for(int j = 0; j < t; j++){
if(i == arr[j] || is(j, arr[j], t, i)) {
flag = 0; break;
}
}
if(flag == 1){
arr[t] = i;
backTrack(t + 1, arr);
arr[t] = -1;
}
}
}
public int totalNQueens(int n) {
m = n;
int[] arr = new int[n];
backTrack(0, arr);
System.out.println(ans);
return ans;
}
}