AcWing寒假每日一题——Day14棋盘挑战

1432. 棋盘挑战

一、问题描述

给定一个$N\times N$ 的棋盘，请你在上面放置 $N$ 个棋子，要求满足：
·每行每列都恰好有一个棋子
·每条对角线上都最多只能有一个棋子

    1   2   3   4   5   6
  -------------------------
1 |   | O |   |   |   |   |
  -------------------------
2 |   |   |   | O |   |   |
  -------------------------
3 |   |   |   |   |   | O |
  -------------------------
4 | O |   |   |   |   |   |
  -------------------------
5 |   |   | O |   |   |   |
  -------------------------
6 |   |   |   |   | O |   |
  -------------------------

上图给出了当 $N$ = 6 时的一种解决方案，该方案可用序列 2 4 6 1 3 5来描述，该序列按顺序给出了从第一行到第六行，每一行摆放的棋子所在的列的位置。
请你编写一个程序，给定一个 $N\times N$ 的棋盘以及 $N$ 个棋子，请你找出所有满足上述条件的棋子放置方案。
输入格式
共一行，一个整数 $N$。
输出格式
共四行，前三行每行输出一个整数序列，用来描述一种可行放置方案，序列中的第 i 个数表示第 i 行的棋子应该摆放的列的位置。这三行描述的方案应该是整数序列字典序排在第一、第二、第三的方案。第四行输出一个整数，表示可行放置方案的总数。
数据范围
6≤$N$≤13
输入样例：

6

输出样例：

2 4 6 1 3 5
3 6 2 5 1 4
4 1 5 2 6 3
4

二、问题分析

该题为n皇后变种问题，核心思想没有变话，dfs搜索找出答案即可。
代码如下（示例）：

#include<iostream>
using namespace std;
int n,cnt;
int q[30];
bool row[30],dg[30],udg[30];
void dfs(int u)
{
	if(u==n)
	{
		cnt++;
		if(cnt<=3)
		{
			for(int i=0;i<n;i++)
			{
				cout<<q[i+1]<<" ";
			}
			cout<<endl;
		}
		return;
	}
	for(int i=0;i<n;i++)
	{
		if(!row[i]&&!dg[u+i]&&!udg[n-u+i])
		{
			row[i]=dg[u+i]=udg[n-u+i]=true;
			q[u+1]=i+1;
			dfs(u+1);
			row[i]=dg[u+i]=udg[n-u+i]=false;
		}
	}
}
int main()
{
	cin>>n;
	dfs(0);
	cout<<cnt<<endl;
	return 0;
}