n皇后 回溯法 c++

        n皇后问题应该是算法中比较有知名度的问题了，在这里再次介绍一下题目：

在一个 n * n 的棋盘上放 n 个皇后，每两个皇后不能在同一行，同一列，同一条斜线上。

         以国际象棋棋盘举例8*8的棋盘上摆放8个皇后，上图是其中之一解，要求输入n作为棋盘大小和皇后数量，输出有多少种摆法。

        用到的是回溯法，思路是写一个is函数来判断有行、列、斜线上没有重复放置皇后 （pos[i]是存放i行的列数）。queen函数是计算和放置皇后的主要函数先要判断当前的row（行）是不是已经到了最后，这是对后面递归做出口并且计数加一，for循环是从左到右遍历每一列。

                ！因为每次递归都是在当前行上从左到右查找放置皇后的地方。这个想法从queen的递归中参数row+1就可以看出来 ！                        这是本题一个point！

---

---

                以下是具体代码

#include<iostream>
#include<cmath>
#include<vector>
using namespace std;
int ans = 0;

bool is(vector <int>& pos, int row, int col)
{
    for (int i = 0; i < row; i++)
    {
        if (row == i || col == pos[i] || abs(row - i) == abs(col - pos[i]))
            return false;
    }
    return true;
}

void queen(int n, int row, vector<int>& pos, int ans1)
{
    if (row == n)
    {
        ans++;
        return ;
    }
    for (int i = 0; i < n; i++)
    {
        if (is(pos, row, i))
        {
            pos[row] = i;
            queen(n, row + 1, pos, ans);
        }
    }
    
}


int main()
{
    
    int n;
    cin >> n;
    vector<int >pos(n, 0);
    int ans1 = 0;
    queen(n,0,pos,ans1);
    cout << ans << endl;
    return 0;
}

---

---

完成以后就可以看这题相似的n皇后了，改变的只不过是把皇后的摆法以二维数组输出。

        is中依旧不需要考虑行重复，原因和上题一致。思路相同，也是在每次递归中寻找每行可以放皇后的地方，row = n时就是最后一行放置皇后，也就是一种情况完成，给二维数组push_back就可以了。

两种输出都可以，迭代器和两个for循环。注释掉一种就可以了。

---

---

                以下是具体代码

#include<vector>
#include<iostream>
#include<string>
#include<list>
#include<algorithm>
using namespace std;
vector<vector<string>> ans;

bool is(vector<string>& queen, int row, int col, int n)
{
	for (int i = 0; i < row; i++)								//row当前最远的行数
	{
		if (queen[i][col] == 'Q')
			return false;
	}
	for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--)
	{
		if (queen[i][j] == 'Q')
			return false;
	}
	for (int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++)
	{
		if (queen[i][j] == 'Q')
			return false;
	}
	return true;
}

void back(vector<string>& queen, int n, int row)
{
	if (row == n)
	{
		ans.push_back(queen);
		return;
	}

	for (int col = 0; col < n; col++)
	{
		if (is(queen, row, col, n))
		{
			queen[row][col] = 'Q';
			back(queen, n, row + 1);
			queen[row][col] = '.';
		}
	}

}

vector<vector<string>> soqueen(int n)
{
	
	vector<string> queen(n, string(n, '.'));
	back(queen, n, 0);
	return ans;
}

int main()
{
	int n;
	cin >> n;
	soqueen(n);
	/*for (int i = 0; i < ans.size(); i++)
	{
		for (int j = 0; j < n; j++)
			cout << ans[i][j];
		cout << "\n";
	}*/

	for (int i = 0; i < ans.size(); i++)						//ans.size()是二维数组的大层数量（外层数量）
																//易得ans[1].size()是二维数组1中的size
	{
		for (vector<string>::iterator it = ans[i].begin(); it != ans[i].end(); it++)
			cout << *it;
		cout << '\n';
	}

	return 0;
}

---

---

小小总结：

第二题比较不错的是把一维数组push_back放到二维数组中，很方便。

但是回溯法的时间还是太差了，时间复杂度是O(n*n!)，用位运算可以比较好的解决这个问题。

（完成第二题的代码才发现在最后的打印输出for循环添加一个count++就可以实现输出具体摆法和摆法的数量了）

看完代码是不是发现，不考虑时间复杂度的话，大名鼎鼎的n皇后问题也不过如此（狗头）