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

最新推荐文章于 2025-03-24 20:47:21 发布

原创最新推荐文章于 2025-03-24 20:47:21 发布 · 550 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#洛谷 #DFS

《算法刷题》专栏收录该内容

33 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索解决八皇后问题的方法，并通过一个具体实例进行了详细解析。该算法能够找出所有符合条件的棋子放置方案，并按字典顺序输出前三个解及解的总数。

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

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

题目链接：洛谷P1219 [USACO1.5]八皇后 Checker Challenge

题目描述

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

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

行号 1 2 3 4 5 6

列号 2 4 6 1 3 5

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

输入格式

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

输出格式

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

输入输出样例

输入 #1

输出 #1

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

说明/提示

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

解题思路：

用dfs递归所有行，对于每一行我们枚举所有列的位置，开辟答案，列，左对角线，右对角线的数组，每一此判断当前位置是否合法，如果合法就将列，左对角线，右对角线的数组进行标记，将当前列的数存进答案数组，然后进入下一行，直到最后一行完成，代表答案数+1，如果为前三就输出对应的答案（存储在答案数组中）

#include <iostream>
using namespace std;
const int maxn = 100;

//记录数组：行，列，左对角线，右对角线
int a[maxn],b[maxn], c[maxn],d[maxn];
int ans = 0, n;

void dfs(int st)
{
    //遍历完所有行为出口
    if(st == n+1)
    {
        ans++;
        //前三个输出
        if(ans <= 3)
        {
            for(int i = 1; i <= n; i++)
            {
                cout << a[i] << " ";
            }
            cout << endl;
        }
        return ;
    }
    //枚举st行中所有列的位置
    for(int i = 1; i <= n; i++)
    {
        //判断当前列位置上，列/左对角线/右对角线是否没有皇后存在
        if(b[i] == 0 && c[st - i + n] == 0 && d[i + st] == 0)
        {
            //记录行的位置
            a[st] = i;
            //标记当前列
            b[i] = 1;
            //标记左对角线
            c[st - i + n] = 1;
            //标记右对角线
            d[i + st] = 1;
            //进入下一行
            dfs(st + 1);
            //将之前的标记清空
            a[st] = 0;
            b[i] = 0;
            c[st - i + n] = 0;
            d[i + st] = 0;
        }
    }
}

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