52.N皇后II(N-Queens II)

版权声明:本文为博主原创文章,未经博主允许不得转载。博主水平有限,欢迎指教 https://blog.csdn.net/Follower_JC/article/details/83996255

题目描述

n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。
在这里插入图片描述
给定一个整数 n,返回 n 皇后不同的解决方案的数量。

解题思路

嗯,这题就是上一题(51.N皇后)的翻版,上一题要求输出所有解法,这一题却只要求数量就行了。
所以。。。当然是选择在51题的基础上套一层皮啦(滑稽)
51题解法参照:
51.N皇后(N-Queens)

实现代码

class Solution {
    public int totalNQueens(int n) {
        boolean[] l = new boolean[n];
        boolean[] x1 = new boolean[2*n-1];
        boolean[] x2 = new boolean[2*n-1];
        
        List<List<String>> list = new ArrayList();
        List<String> tmp = new ArrayList();
        
        if(n==0)    return 0;
        
        find(list, tmp, n, l, x1, x2);
        
        return list.size();
    }
    
    public static void find(List<List<String>> list, List<String> tmpList, int n, boolean[] l, boolean[] x1, boolean[] x2){
        if(tmpList.size()==n){
            list.add(new ArrayList<>(tmpList));
            return;
        }
        int i = tmpList.size();
        for(int j=0;j<n;j++){
            if(l[j]==false && x1[j+i]==false && x2[n-1+i-j]==false){
                tmpList.add(append(j, n));
                l[j]=true;
                x1[j+i]=true;
                x2[n-1+i-j]=true;
                find(list, tmpList, n, l, x1, x2);
                l[j]=false;
                x1[j+i]=false;
                x2[n-1+i-j]=false;
                tmpList.remove(tmpList.size()-1);
            }
        }

    }
    
    public static void println(List<String> list){
        for(int i=0;i<list.size();i++)
            System.out.print(list.get(i)+" ");
        System.out.println();
        return;
    }
    
    public static String append(int ind, int n){
        StringBuilder res = new StringBuilder("");
        for(int i=0;i<ind;i++){
            res.append(".");
        }
        res.append("Q");
        for(int i=ind+1;i<n;i++){
            res.append(".");
        }
        return res.toString();
    }
}

猜你喜欢

转载自blog.csdn.net/Follower_JC/article/details/83996255