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

传送门：P1219 [USACO1.5] 八皇后 Checker Challenge

题目描述

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

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

行号 1 2 3 4 5 6

列号 2 4 6 1 3 5

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

输入格式

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

输出格式

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

输入输出样例

输入 #1

6

 输出 #1

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

 说明/提示

【数据范围】
对于 100% 的数据， 。

题目翻译来自NOCOW。

USACO Training Section 1.5 

----------------------------------------------------------分割线--------------------------------------------------------------

知识点

深度优先搜索（即DFS）

解题思路

由题目得，由于每一行，列只能放一个棋子，每条对角线（包括两条主对角线的所有平行线）上至多有一个棋子，故每个棋子的周围8个点不能再放别的棋子。从第一行开始搜索，只需要着重考虑左下和右下两个点即可。观察样例可得，所有棋子在平移后可以连成一条直线，故设置数组l，r（分别表示左边和右边两个方位）。对于左边的格子，我们观察发现，左下的方位x+y为定值，而对于右边，y-x为定值，为了防止y-x<0，我们在此基础上加n，即将该点向右平移n个单位。通过不断判断该方格是否可以放置棋子，完成搜索可行解的过程。此外，使用1维数组而不是2维数组，也减少了循环的次数，缩短了搜索的复杂度，避免重复搜索。

AC代码

#include <iostream>
#include <algorithm>
using namespace std;
int n,ans=0;
int solution[100],p[100],l[100],r[100];//l为左下（left)，r为右下(right)
void dfs(int y) {
    if (y == n + 1) {//全部搜索完毕
        if (ans <= 2) {//ans=0时输出一组，ans=1时输出一组，ans=2时输出一组，一共3组
            for (int i = 1; i <= n; i++)
                cout << solution[i] << " ";
            cout << endl;
        }
        ans++;
        return;//返回上一次dfs
    }
    for (int i = 1; i <= n; i++) {
        if (p[i] == 0 && l[y + i] == 0 && r[i - y + n] == 0) {
            solution[y] = i;//第y行的横坐标i
            p[i] = 1;
            l[y + i] = 1;
            r[i - y + n] = 1;//标记该点为已占领

            dfs(y + 1);//搜索下一行的可行解
            p[i] = 0;//未找到，将刚占领的点重置为未占领
            l[y + i] = 0;
            r[i - y + n] = 0;;
        }
    }
}
int main() {
    cin >> n;
    dfs(1);//从第一行开始搜索
    cout << ans << endl;
    return 0;
}