LeetCode 52. N皇后 II

题目链接:

力扣icon-default.png?t=M276https://leetcode-cn.com/problems/n-queens-ii/LeetCode 52. N皇后 II_第1张图片

【分析】回溯法的经典例题,通过一个数组来存储棋子的位置,数组下标为行号,存的值为列号,在改变数组值之前进行判定。

需要注意用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;
    }
}

 

你可能感兴趣的:(LeetCode,leetcode,回溯法)