leetcode之NQueue

#include "stdafx.h"
#include<vector>
using namespace std;
/*
  n皇后问题的思路：n皇后即n个皇后分布于不同的行、列，且两两皇后都不在同一对角线，解题思路是从0到n-1逐行扫描，针对每一行，首先找到没有皇后的列，通过行列定位一个点，然后针对当前点，判断其与已知的皇后是否处于同一对角线，若否，将其写入，然后继续遍历下一层，当扫描完最后一行时，找到一组解，由于每行在寻找列时都是遍历所有的列，所以递归结束时所有情况都已找到。
*/
vector<vector<string>>vv;//用于存放最终结果
int colFlag[100];//标记当前列是否有皇后，1表示有，0表示没有
int rowCol[100];//标记下标行的皇后所在的列，即通过皇后所在的行数查找其列数
class Solution{
private:
	
public:
	vector<vector<string>> solveNQueens(int n){

		if( n == 0)
			return vv;
		DFS(0,n);
		return vv;	    
	}
	void DFS(int row,int n){
		if(row == n){
			vector<string>v;
			for(int i = 0; i < n; i++){
			    string temp(n,'.');
			    temp[rowCol[i]] = 'Q';		
				v.push_back(temp);
			}
			vv.push_back(v);
			return;//递归结束
		}
		//如果row != n
		for(int j = 0; j < n; j++){// 对所有列进行遍历，每次遍历可能有多个列满足条件，要将所有条件的都进行实验
			if(colFlag[j] == 0){  //i为没有标记皇后的列，下面对当前点(row，j)判断对角线是否满足条件
				int i = 0;  //i依次扫描前row行
				for(i = 0; i < row; i ++){//标记过皇后的行数位于row之前,皇后所在点（i,rowCol[i]）
					if(abs(row - i) ==abs(j - rowCol[i]))
						break;
				}
				if(i == row){//此时满足插入皇后的条件
					colFlag[j] = 1;
					rowCol[i] = j;
					DFS(row+1,n);
					colFlag[j] = 0;
					rowCol[i] = 0;
				}
			}
		}

	}

};

注：n皇后中的递归比较复杂，是一个典型的回溯问题，在判断对角线问题时应特别注意是对应的行值减行值，对应的列值减列值。另外，在传参过程中，如果将结果作为参数传递，要采用传地址的方式（&）