acwing 843. n-皇后问题

目录

题目描述

n−皇后问题是指将 n 个皇后放在 n×n 的国际象棋棋盘上,使得皇后不能相互攻击到,即任意两个皇后都不能处于同一行、同一列或同一斜线上。

acwing 843. n-皇后问题

现在给定整数 n,请你输出所有的满足条件的棋子摆法。

输入格式

共一行,包含整数 n。

输出格式

每个解决方案占 n 行,每行输出一个长度为 n 的字符串,用来表示完整的棋盘状态。

其中 . 表示某一个位置的方格状态为空,Q 表示某一个位置的方格上摆着皇后。

每个方案输出完成后,输出一个空行。

注意:行末不能有多余空格。

输出方案的顺序任意,只要不重复且没有遗漏即可。

数据范围

1≤n≤9

输入样例:

4

输出样例:

.Q..
...Q
Q...
..Q.

..Q.
Q...
...Q
.Q..

dfs算法求解

分析

dfs顺序很重要

采用全排列的dfs模板,用每一层u代表棋盘的每一行

然后在每一层dfs里面枚举每一列 i

画个图可以得到每个正对角线的编号分别为 u+i

副对角线的编号为i - u,为了防止出现负数,整体加上n,所以是i-u+n

acwing 843. n-皇后问题

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

const int N = 10;

char g[N][N]; 
int n;
bool col[N], dg[N], udg[N]; // 用于标记每一列、正对角线、负对角线是否占用

// u 表示第u行 
void dfs(int u)
{ 
	if(u == n)
	{
		for(int i = 0; i < n; i++) puts(g[i]);
		puts(""); 
		return ;	
	}	
	
	// i 表示第i列 
	for(int i = 0; i < n; i++)
	{
		// u表示行,i表示列,u+i表示正对角线的编号,i-u+n表示负对角线编号 
		if(!col[i] && !dg[u+i] && !udg[i-u+n])
		{
			g[u][i] = 'Q'; // 都不冲突的话,就把一个皇后放到这里 
			col[i] = dg[u+i] = udg[i-u+n] = true; // 表示该列、对角线被占用 
			dfs(u+1); //下一行
            
			g[u][i] = '.'; 
			col[i] = dg[u+i] = udg[i-u+n] = false; // 
			
		}
	}
}




int main()
{
	scanf("%d", &n);
	// 初始化g
	for(int i = 0; i < n; i++) 
		for(int j = 0; j < n; j++)
			g[i][j] = '.';
	
	dfs(0);
	
	return 0;
}

时间复杂度

参考文章

https://www.acwing.com/activity/content/code/content/47097/

上一篇:洛谷P2392 kkksc03考前临时抱佛脚


下一篇:cf883 G. Orientation of Edges(贪心)