dfs学习之八皇后问题

1、题目大意

[USACO1.5] 八皇后 Checker Challenge

题目描述：

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

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

行号 $123456$

列号 $246135$

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

输入格式：

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

输出格式：

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

样例 #1

样例输入 #1

6

样例输出 #1

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

提示

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

2、知识点部分

2.1 矩阵对角线特点及标记方法

在任意矩阵内设一点D的坐标为：(i , j) 那么其左下角，或者右上角的点Y (x , y)，满足i + j = x + y = 一个常数，同时，其左上角和右下角的点X (x , y)，满足 x - y + n = i - j + n = 一个常数,这里的n一般是一个常数，相当于一个偏移量，防止当x>y或者x>=y的时候导致开的数组下标<=0（根据题目来定，该题我选择从数组下标1开始，所以判断数组要开的足够大，一般对角线长度至少为：2 * n - 1）。
那么标记该位置是否已经访问过这个问题已经解决，现在我们需要解决第二个问题就是回溯，回溯实际上就是在每一步dfs退回到这个点继续进行下一步的时候维护（恢复）原始状态。
dfs关键的地方在于找到关键的返回值，即何时碰到“墙”并返回，并且记住每一步走过的路，回溯的话就是不管这条路能不能走，我们都要在走完之后恢复原样，但因为是走完之后，所以在我走一条路走到黑之前是不需要进行回溯的，所以回溯语句一般出现在递归调用dfs之后。

3、代码如下

#include<iostream>

using namespace std;
const int maxn = 15;
int path[maxn];
int st[maxn],udg[maxn],dg[maxn];
int n,ans;
void dfs(int x)
{
    if(x > n)
    {
        if(ans<=2)
        {
            for(int i = 1;i<=n;i++)
            {
                if(i == n) cout<<path[i]<<endl;
                else cout<<path[i]<<' ';
            }
        }
        ans++;
        return;
    }

    for(int i = 1;i<=n;i++)
    {
        if(st[i] || udg[x+i] || dg[x - i + n]) continue;
        else{
            path[x] = i;
            st[i] = udg[x+i] = dg[x - i + n] = 1;
            dfs(x+1);
            st[i] = udg[x+i] = dg[x - i + n] = path[x] = 0;
        }
    }
}
int main()
{
    cin>>n;
    dfs(1);
    cout<<ans;
    return 0;
}