棋盘挑战【DFS】

棋盘挑战【DFS】

题目链接
给定一个 N×N 的棋盘，请你在上面放置 N 个棋子，要求满足：

每行每列都恰好有一个棋子
每条对角线上都最多只能有一个棋子

上图给出了当 N=6 时的一种解决方案，该方案可用序列 2 4 6 1 3 5 来描述，该序列按顺序给出了从第一行到第六行，每一行摆放的棋子所在的列的位置。

请你编写一个程序，给定一个 N×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

题解：对于一条斜率为正的（shang）对角线：y=x+b，截距为b=y-x,遍历过程难免出现y<x的情况，所以加上一个棋盘尺寸n,让b=n+y-x，这样b恒正,我们在处理这个恒正的下标时就方便许多。当然，对于xia对角线b=y+x无需多虑。
建立一个ans数组，去记录每一行的棋子是出现在第几个位置上的。用cnt去记录前三次符合题意的方案，并将符合题意的方案输出。
开始的时候建立三个bool类型数组用作判断。

bool lie[N], shang[N], xia[N];

//全局bool类型 默认为false

分别代表列，和两条斜率不同的对角线。
从x=0开始搜索，遇到符合条件的位置就把位置用ans数组记录下来，此时把对应的下标的列数组，对角线数组标记为true，然后dfs(x+1)。下一次遇到列和对角线数组都不为true的时候（代表可以放棋子）再记录位置，并标记位置，进行递归。
当前循环标记之后记得回溯之前的状态，不要影响上一层循环。

#include <iostream>
using namespace std;
const int N = 50;
int n,cnt=0;//cnt默认为0；
int ans[N];//存要输出的答案位置 
bool lie[N], shang[N], xia[N];//shang是斜率为正对角线，xia是斜率为负对角线，lie是代表列 。 
void dfs(int x)//以x递增的方式深搜，全局按列搜索 
{
    if (x ==n)//遍历完成
    {   cnt++; 
		if(cnt<=3)
    {
     	for(int i=0;i<n;i++)
     	{
     		if(i==0) printf("%d",ans[i]);
     		else  printf("% d",ans[i]);
		 }
		 printf("\n");    	  
	}	   
    }
      
    for (int i = 0; i < n; i ++ )//对某一列，按行搜索 
        if (!lie[i] && !xia[x + i] && !shang[n - x + i]) //三个false 为合法  
        {
            ans[x]=i+1;
            lie[i] = xia[x + i] = shang[n - x + i] = true;//标记 
            dfs(x + 1);
            lie[i] = xia[x + i] = shang[n - x + i] = false;//恢复上一层初态 
        }
}

int main()
{
    cin >> n;
    dfs(0);
    cout<<cnt<<endl;
    return 0;
}   

类似题解：n皇后问题（基本上一个算法）
n皇后问题