Leetocde51: 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.

For example,
There exist two distinct solutions to the 4-queens puzzle:

[
 [".Q..",  // Solution 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // Solution 2
  "Q...",
  "...Q",
  ".Q.."]
]

题意：

经典的N皇后问题，N皇后问题的要求是棋盘上每一行每一列以及每一个斜线上都不能同时出现两个皇后。

解法一：

思路：是在看了别的博主po的文章后效仿，将原先一般用来表示棋盘的二维数组简化为一维数组，a[i]=j 表示的意思是，第 i 行的皇后放在 j 列上，也就是 i,j 位置处。
N皇后问题是回溯的典型，下面使用的是非递归的方式。
从第一行开始，选择皇后的位置，每一步能不能放皇后用一个判断函数进行判断，对于一维数组表示的棋盘，不存在行重复的问题，需要判断的就是列重复和斜线重复。列重复就是遍历是否已经出现过相同的列，斜线则是横纵坐标的关系。利用该判断函数判断当前的位置能不能下。如果不能下的话，就回溯，将上一步下的皇后的位置进行改动，一直移动到改动第一行的皇后的位置也没解的时候，说明所有的解都找到了。如果能下，而且在最后一行，就将该次寻找的结果保存起来。但是不能立即结束，而是要继续改变最后一行的皇后的位置，重复上面的过程，不然找到的就不是全部的解。

代码：

<span style="font-size:14px;">class Solution {
public:
    int mmin=-100000;

	bool valid(int i, int j, int a[], int n){
		for(int r=0; r<n; r++){
			if(a[r]==j || abs(r-i)==abs(a[r]-j)){
				return false;
			}
		}
		return true;
	}
	
	vector< vector<int> > queen(int n){
		vector< vector<int> > ans;
		int a[n];
		for(int i=0; i<n; i++)
		a[i]=mmin;
		int i=0, j=0;
		while(i<n){
			while(j<n){
				if(valid(i, j, a, n)==true){
					a[i]=j;
					j=0;
					break;
				}
				else{
					j++;
				}
			}
			if(a[i]==mmin){
				if(i==0) break;
				else{
					i--;
					j=a[i]+1;
					a[i]=mmin;
					continue;
				}
			}
			if(i==n-1){
				vector<int> va;
				for(int r=0; r<n; r++) 
				va.push_back(a[r]);
				ans.push_back(va);
				j=a[i]+1;
				a[i]=mmin;
				continue;
			}
			i++;
		}
		return ans;
	}
	
	
    vector< vector<string> > solveNQueens(int n) {
        vector< vector<int> > ans = queen(n);
        vector< vector<string> > res;
        int aLen=ans.size();
        for(int i=0; i<aLen; i++){
        	vector<string> vr;
        	string str(n, '.');
        	for(int j=0; j<n; j++){
        		str[ans[i][j]]='Q';
        	    vr.push_back(str);
        	    str[ans[i][j]]='.';
        	}
			res.push_back(vr);
		}
		
		return res;
    }
};</span>