52. N-Queens II

Follow up for N-Queens problem.

Now, instead outputting board configurations, return the total number of distinct solutions.

这是n皇后问题的第二种问题形式,我们在上一题中将所有放置皇后的方式都找了出来并存储在集合中,现在,我们问题貌似变得简单了,只需要计算出总共有多少中放置皇后的方式即可。事实上,我们可以根据上一题的结果,直接计算集合中的元素个数就是本题的解,但是这显然是一种效率低下的方法,因为相对于本题来说,我们会做很多无用功。

上一题的链接:点击打开链接

那么,我们需要做的是,怎样尽可能的优化上一题的解法,并且达到计算此题的目的(参考了讨论区的思路)。

首先,我们注意到上一题中需要集合来存储每一种放置皇后的方式,而现在显然不需要了,只要一个变量用来存储次数即可。

第二,在此题中,我们不需要知道皇后具体存放在什么位置,只需要知道哪一行,哪一列,以及哪一条斜线已经放置了皇后就行了。总共有n行n列,

所以定义一个元素为n的boolean数组即可,用来表示第i列是否已经放置了皇后。又,左上到右下的斜线和左下到右上的斜线都有2n-1条,所以定义一个元素为2n-1的boolean数组即可,用来表示某一条斜线是否已经放置了皇后。

基于上述2点,我们在上一题的解法下做了如下优化,解决了此题。

下面是代码:

public class Solution {
    int count = 0;
    public int totalNQueens(int n) {
        boolean[] b1 = new boolean[n];         //第i列是否放置皇后
        boolean[] b2 = new boolean[2*n-1];     //第i条斜线(左上到右下)是否放置皇后
        boolean[] b3 = new boolean[2*n-1];     //第i条斜线(左下到右上)是否放置皇后

        fon(0,b1,b2,b3,n);
        return count;
    }
    public void fon(int row, boolean[] b1, boolean[] b2, boolean[] b3 , int n){
        if(row == n) {
            count++;
            return;
        }
        for(int i=0;i

这种方法在LeetCode上超越了87.87%,是一种不错的解法!这2道题主要用到的是回溯算法,回溯算法的一种形象比喻就是: 我们去爬山有很多上山的入口,每个入口都有很多岔路,我们任意选择一个路口前行,当到达山顶或者遇到死路时,退回到上一个路口,再重新选择一条路前进。




你可能感兴趣的:(leetcode)