n皇后问题研究java实现

题目描述

https://leetcode-cn.com/problems/n-queens

n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

上图为 8 皇后问题的一种解法。

给定一个整数 n，返回所有不同的 n 皇后问题的解决方案。

每一种解法包含一个明确的 n 皇后问题的棋子放置方案，该方案中 'Q' 和 '.' 分别代表了皇后和空位。

示例:

输入: 4
输出: [
 [".Q..",  // 解法 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // 解法 2
  "Q...",
  "...Q",
  ".Q.."]
]
解释: 4 皇后问题存在两个不同的解法。

思路：要找出所有可能的解法，因此想到用回溯，DFS， 每行一个Q，所以逐行遍历

判断该位置是否能放Q 取决于它所在的列、 左对角线和右对角线上是否有Q，用三个boolean型数组存放所有列、 左对角线和右对角线上是否有Q的信息

n*n的矩阵，共有n列，2*n-1条左对角线和2*n-1条左对角线

import java.util.*;
class Solution {
    boolean[] col=null;
    boolean[] left=null;
    boolean[] right=null;
    public List<List<String>> solveNQueens(int n) {
        col=new boolean[n];
        left=new boolean[2*n-1];
        right=new boolean[2*n-1];
        List<List<String>> res=new ArrayList<List<String>>();
        if(n==0){
            List<String> list=new ArrayList<String>();
            res.add(list);
            return res;
        }
        if(n==1){
            List<String> list=new ArrayList<String>();
            list.add("Q");
            res.add(list);
            return res;
        }
        String[][] num=new String[n][n];//存放字符的矩阵先全填入"."
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                num[i][j]=".";
            }
        }
        find(num,0,n,res);//从第0行开始求解
        return res;
    }
    public void find(String[][] num,int start,int n,List<List<String>> res){
        if(start==n){//当n行都求解完成后，输出每行的结果
            List<String> list=new ArrayList<String>();
            for(int i=0;i<n;i++){
                String temp="";
                for(int j=0;j<n;j++){
                    temp=temp+num[i][j];
                }
                list.add(temp);
            }
            res.add(list);
            return;
        }
        for(int i=0;i<n;i++){//每一行都要遍历完所有项
            //若当前位置没有在其他Q的列、左对角线、右对角线上则执行
            if(!col[i]&&!right[start-i+n-1]&&!left[start+i]){
                num[start][i]="Q";//当前位置置Q
                col[i]=true;//当前Q的所在列、左右对角线都标记
                right[start-i+n-1]=true;
                left[start+i]=true;
                find(num,start+1,n,res);//递归调用找下一行
                num[start][i]=".";//执行完递归要把当前位置复原，以便判断下一个位置
                col[i]=false;
                right[start-i+n-1]=false;
                left[start+i]=false;
            }
        }
    }
}