洛谷P1219 [USACO1.5]八皇后 Checker Challenge

一个如下的 6×6 的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行、每列有且只有一个，每条对角线（包括两条主对角线的所有平行线）上至多有一个棋子。

上面的布局可以用序列 2 4 6 1 3 52 4 6 1 3 5 来描述，第 i 个数字表示在第 i 行的相应位置有一个棋子，如下：

行号 1 2 3 4 5 6

列号 2 4 6 1 3 5

这只是棋子放置的一个解。请编一个程序找出所有棋子放置的解。
并把它们以上面的序列方法输出，解按字典顺序排列。
请输出前 3 个解。最后一行是解的总个数。

输入格式

一行一个正整数 n，表示棋盘是 n×n 大小的。

输出格式

前三行为前三个解，每个解的两个数字之间用一个空格隔开。第四行只有一个数字，表示解的总数。

输入输出样例

输入 #1

6

输出 #1

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

说明/提示

【数据范围】
对于 100%100% 的数据，6≤n≤13。

思路在下面的代码里面

AC代码

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
typedef long long ll;
using namespace std;
int n,mp[53][53],ans;
int a[53],l[53],r[53],cnt[53];
void dfs(int x)
{
    if( x==n+1 ) //因为下面进入每个搜索行数都要+1.所以终止条件
    {
        ans++;
        if(ans<=3)
        {
            for(int i=1;i<=n;++i) cout<<cnt[i]<<" ";
            cout<<endl;
        }
        return ;
    }


    //因为每行、每列、每条对角线（包括平行对角线）只能有一个棋子
    // 每列一个棋子 ： !a[i]
    // 左对角线(包括平行)规律 ：行数-列数（范围） = [-(n-1),n-1] 因为是下标 所以+n -->  [1,2*n-1]
    // 右对角线(包括平行)规律 ：行数+列数（范围） = [2,2*n]



    for(int i=1;i<=n;++i) // 表示列数i
    {

        if( !a[i] && !l[x-i+n] && !r[x+i] ) // 列没放，左斜没放，右斜没放
        {
            a[i] = l[x-i+n] = r[x+i] = 1;
            cnt[x] = i ;                   // 记录x行的棋子房子
            dfs(x+1);
            a[i] = l[x-i+n] = r[x+i] = 0 ;
        }


    }

}
int main()
{
    scanf("%d",&n);
    dfs(1);  //用深搜做 从第1行开始
    cout<<ans<<endl;


    return 0;
}