LeetCode 051 N-Queens

题目

---

The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.

Given an integer n, return all distinct solutions to the n-queens puzzle.

Each solution contains a distinct board configuration of the n-queens' placement, where 'Q' and '.' both indicate a queen and an empty space respectively.

列出n皇后问题的所有可能性

思路

---

1 经典的递归回溯问题，网上的写法有很多，但思路应该很清楚。

2 关键是 a 数据结构的选取，b 判断的写法 ， c 回溯如何处理

3 我选择了一个数组来作为解决这道题目的问题关键，这个数组代表的含义是a[row]=col,皇后在第row的第col列。好处如下：

a 比二维数组记录内存小

b 判断的时候，简单。b.1 不同行的如果在同列 b.2 不同行如果在同一个斜列。都很容易写出

c 回溯的时候不需要特意恢复这个数组，因为下次进行的时候，直接修改此值。

4 这个题目还是自己写几遍，才会清楚

代码

---

public class Solution {
    public ArrayList<String[]> solveNQueens(int n) {
        ArrayList<String[]> ans = new ArrayList<String[]>();
        if(n==0){
            return ans;
        }
        int[] rcol = new int[n];
        queens(ans,rcol,0,n);
        return ans;
    }
    
    public void queens(ArrayList<String[]> ans ,int[] rcol,int row,int n){
        if(row==n){
            printqueen(ans,rcol,n);
            return;
        }
        for(int col=0;col<n;col++){
            rcol[row]=col;
            if(check(rcol,row)){
                queens(ans,rcol,row+1,n);
            }
        }
    }
    
    public boolean check(int[] rcol,int row){
        for(int i=0;i<row;i++){
            if(rcol[i]==rcol[row]){
                return false;
            }
            if(Math.abs(rcol[i]-rcol[row])==row-i){
                return false;
            }
        }
        return true;
    }
    
    public void printqueen(ArrayList<String[]> ans ,int[] rcol,int n){
        String[] str = new String[n];
        for(int i=0;i<n;i++){
            String temp = new String();
            for(int j=0;j<n;j++){
                if(rcol[i]==j){
                    temp+="Q";
                }
                else{
                    temp+=".";
                }
            }
            str[i]=temp;
        }
        ans.add(str);
    }
}