【洛谷P1219】八皇后

题目描述

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

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

行号 1\ 2\ 3\ 4\ 5\ 6

        1 2 3 4 5 6

列号 2\ 4\ 6\ 1\ 3\ 5

        2 4 6 1 3 5

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

输入格式

一行一个正整数 nn，表示棋盘是 n \times nn×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 \le n \le 136≤n≤13。

题目翻译来自NOCOW。

USACO Training Section 1.5

AC代码如下：

#include<cstdio>
#include<cmath>
#include<iostream>
using namespace std;
int a[15]= {0};//下标存储行 元素存储列
int sum=0;//次数
int n;
int tag[15]= {0};//标记一下列是否被使用过

int check(int x)//查看斜线
{
    for(int i=1; i<x; i++)
    {
        if((x-i)==abs(a[x]-a[i]))
        {
            return 0;
        }
    }
    return 1;
}
void dfs(int x)
{
    if(x>=n+1)
    {

        sum++;

        if(sum<=3)//由于是列从小到大进行尝试 所以一定会按字典序输出
        {
            for(int i=1; i<=n; i++)
            {
                if(i!=n)
                    printf("%d ",a[i]);
                else
                    printf("%d",a[i]);

            }
            printf("\n");
        }
        return;
    }
    for (int i = 1; i <= n; i++)
    {
        if (tag[i] == 0 )
        {
            tag[i] = 1;
            a[x] = i;//重点 需要先给a[x]存储进行 才可以进行check 否则a[x]=0
            if (check(x) == 1)


            {
                dfs(x + 1);
                
               
            }
            tag[i] = 0;//回溯时要 抹去列的标记

        }

    }
    return;
}
int main()
{
    cin>>n;
    dfs(1);
    cout<<sum;
    return 0;

}

 通过图