n-皇后问题

n-皇后问题

#include<bits/stdc++.h>
using namespace std;
const int N=510;
int n;//表示行数
char g[N][N];
bool col[N],dg[N],udg[N];//分别存储该列，对角线，反对角线是否有皇后
void dfs(int u){
	if(u==n+1){//已经搜索了n行
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				cout<<g[i][j];
			}
			cout<<endl;
		} 
		cout<<endl;
		return ;
	}//找到相应的解进行输出
	for(int i=1;i<=n;i++){//遍历每一列
		if(!col[i]&&!dg[i-u+n]&&!udg[i+u]){//约束性条件，进行剪枝
			 col[i]=dg[i-u+n]=udg[i+u]=true;
			 g[u][i]='#';
			 dfs(u+1);//搜索下一行
			 g[u][i]='.';
			 col[i]=dg[i-u+n]=udg[i+u]=false;//进行回溯
		}
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			g[i][j]='.';
		}
	}
	dfs(1);
	return 0;
}//题目采用acwing中的n皇后问题