LeetCode 51. N 皇后 java题解

最新推荐文章于 2021-03-10 05:27:21 发布

奔跑的废柴

最新推荐文章于 2021-03-10 05:27:21 发布

阅读量178

点赞数

分类专栏： LeetCode 文章标签：深度优先搜索 N皇后问题回溯法冲突检查算法解析

本文链接：https://blog.csdn.net/weixin_42970433/article/details/112027777

版权

LeetCode 专栏收录该内容

279 篇文章 6 订阅

订阅专栏

题目

https://leetcode-cn.com/problems/n-queens/
在这里插入图片描述

分析

是否在一条斜线上想半天都不对，哭
在这里插入图片描述

代码

class Solution {
    ArrayList<List<String>> list=new ArrayList<>();
    char[][] res;
    int[] column;
    int[] slash1;
    int[] slash2;
    int n;
    public List<List<String>> solveNQueens(int n) {
        if(n==0)
            return new ArrayList<>();
        this.n=n;
        column=new int[n];//列
        slash1=new int[2*n];//行下标与列下标之和相等
        slash2=new int[2*n];//行下标与列下标之差相等
        res=new char[n][n];
        //res初始化为'.'
        for(int i=0;i<n;i++)
            Arrays.fill(res[i],'.');
        //从第一行的第一个皇后开始
        dfs(0);
        //结果集
        /*
        ArrayList<List<String>> listlist=new ArrayList<>();
        for(int i=0;i<list.size();i++){
            char[][] c=list.get(i);
            ArrayList<String> as=new ArrayList<>();
            for(int a=0;a<c.length;a++){
                as.add(String.valueOf(c[a]));
            }
            listlist.add(as);
        }
        return listlist;
        */
        return list;
    }
    //判断第k个皇后放在第几列
    public void dfs(int k){
        if(k==n){
            /*
            char[][] good=new char[n][n];
            for(int i=0;i<n;i++){
                for(int j=0;j<n;j++){
                    good[i][j]=res[i][j];
                }
            }
            list.add(good);//这边一定要是全局res的拷贝，而不是res数组本身
            */
            
            //将res转换格式，加入最终结果中
            ArrayList<String> as=new ArrayList<>();
            for(int a=0;a<res.length;a++){
                as.add(String.valueOf(res[a]));
            }
            list.add(as);
            return;
        }
        for(int i=0;i<n;i++){
            //这一列放不了，因为列中或斜线中已有皇后
            if(column[i]==1||slash1[i+k]==1){
                continue;
            }
            //斜线有皇后
            //差可能为负数，统一加上n
            if(slash2[k-i+n]==1){
                continue;
            }
            else{
                res[k][i]='Q';
                column[i]=1;
                slash1[i+k]=1;
                slash2[k-i+n]=1;
                dfs(k+1);
                res[k][i]='.';
                column[i]=0;
                slash1[i+k]=0;
                slash2[k-i+n]=0;
            }
        }
    }
}