N皇后问题

leetcode51题

题目描述

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

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

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

 

class Solution {
private:
	vector<vector<string>> res;
	vector<bool> cols;
	vector<bool> dis1;
	vector<bool> dis2;
	vector<string> getBoard(int n, vector<int> row)
	{
		vector<string> board(n, string(n,'.'));
		for (int i = 0; i < n; i++)
		{
			board[i][row[i]] = 'Q';
		}
		return board;
	}

	void putQueen(int n, int index, vector<int>& row)
	{
		if (index == n) {
			res.push_back(getBoard(n,row));
		}
		for (int i = 0; i < n; i++)
		{
			if (!cols[i] && !dis1[i+index] && !dis2[index-i+n-1])
			{
				row.push_back(i);
				cols[i] = true;
				dis1[i + index] = true;
				dis2[index - i + n - 1] = true;
				putQueen(n, index + 1, row);
				cols[i] = false;
				dis1[i + index] = false;
				dis2[index - i + n - 1] = false;
				row.pop_back();
			}
		}
		return;
	}

public:
	vector<vector<string>> solveNQueens(int n) {
		res.clear();
		cols = vector<bool>(n, false);
		dis1 = vector<bool>(2*n-1, false);
		dis2 = vector<bool>(2 * n - 1, false);
		vector<int> row;
		putQueen(n, 0, row);
		return res;
	}
};