n皇后问题（dfs+剪枝）

原题：传送门
题意： 给n x n的棋盘，放n个皇后，要求任何两个皇后不能在同一行或同一列或同一对角线上，按字典序输出所有方案
思路： 用ans数组存储第i行皇后所在的列，这样就不存在两个皇后在同一行的情况了；用vis[i]记录第i列是否存在皇后，这样就不存在两个皇后在同一列的情况了；再写个check(int m)检查当前摆法的对角线的合法性，dfs搜索就完事了
友情提示： dfs()返回后，一定要把被置为true的vis[i]还原为false
Code：

#include
using namespace std;

int n;
int ans[17];//第i行存储皇后所在位置ans[i]
bool vis[17];//vis[i]表示数字i是否被使用过，也就是第i列是否被摆放了皇后
bool is_print = false;

bool check (int m) {
     
	for (int i=m+1;i<=n;i++)//只判断前m个位置,故把m+1到n得数值清零
		ans[i] = 0;
	
	for (int i=1;i<=m;i++) {
     
		for (int j=1;j<=ans[i] - 1;j++) {
     
			if (ans[i] - j == ans[i - j] || ans[i] - j == ans[i + j]) {
     
				return false;
			}
			
		}
		for (int j=1;j+ans[i]<=m;j++) {
     
			if (ans[i] + j == ans[i - j] || ans[i] + j == ans[i + j]) {
     
				return false;
			}
		}
	}
	return true;
}
void dfs (int m) {
     
	if (m-1 == n) {
     //因为从1开始的dfs,所以取n个值时,m已经等于n+1了
		is_print = true;
		for (int i=1;i<=n;i++) {
     
			cout<<ans[i];
			if (i != n) {
     
				cout<<" ";
			} else {
     
				cout<<endl;
			}
		}
	}
	for (int i=1;i<=n;i++) {
     
		if (!vis[i]) {
     
			ans[m] = i;
			if (!check(m)) {
     //当前m个都不满足时,就没必要继续了,实现了所谓的剪枝
				ans[m] = 0;
				continue;
			}
			vis[i] = true;
			dfs(m+1);
			vis[i] = false;
		}
	}
}
int main()
{
     
	cin>>n;
	fill(vis, vis + 17, false);
	dfs(1);
	if (!is_print) {
     
		cout<<"no solute!"<<endl;
	}
	return 0;
}