USACO 1.5.4 Checker Challenge跳棋的挑战(回溯法求解N皇后问题+八皇后问题说明)

 Description

检查一个如下的6 x 6的跳棋棋盘,有六个棋子被放置在棋盘上,使得每行,每列,每条对角线(包括两条主对角线的所有对角线)上都至多有一个棋子。 列号

0   1   2   3   4   5   6
  -------------------------
1 |   | O |   |   |   |   |
  -------------------------
2 |   |   |   | O |   |   |
  -------------------------
3 |   |   |   |   |   | O |
  -------------------------
4 | O |   |   |   |   |   |
  -------------------------
5 |   |   | O |   |   |   |
  -------------------------
6 |   |   |   |   | O |   |
  -------------------------

上面的布局可以用序列2 4 6 1 3 5来描述,第i个数字表示在第i行的相应位置有一个棋子,如下: 行号 1 2 3 4 5 6 列号 2 4 6 1 3 5 这只是跳棋放置的一个解。请遍一个程序找出所有跳棋放置的解。并把它们以上面的序列方法输出。解按字典顺序排列。请输出前3个解。最后一行是解的总个数。 特别注意: 对于更大的N(棋盘大小N x N)你的程序应当改进得更有效。不要事先计算出所有解然后只输出,这是作弊。如果你坚持作弊,那么你登陆USACO Training的帐号将被无警告删除

Input

一个数字N (6 <= N <= 13) 表示棋盘是N x N大小的。

Output

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

Sample Input

6

Sample Output

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


解题思路:求解n皇后的裸题。
之前就一直想写写关于八皇后问题的博客,这次碰到裸题,就一起说了吧。

八皇后问题:

在棋盘上放置8个皇后,使得它们互不攻击,此时每一个皇后的攻击范围为同行同列和同对角线,要求找出所有的解。

分析:最简单的思路是把问题转化为"从64个格子中选取一个子集",使得“子集中掐有8个格子,且任意两个选出的格子都不在同一行、同一列、同一个对角线上”,这正是子集的枚举问题,
然而64个格子的子集有2^64个,太大了,这并不是一个很好的模型。
第二个思路是把问题转化为“从64个格子中选8个格子”,这是组合生成问题。根据组合数学,有4.426*10^9种方案,比第一种优秀,但仍然不够很好。
经过思考,不难发现:恰好每行每列各放置一个皇后,如果用C[x]表示第x行皇后的编号,则问题变成了全排列生成问题。而0-7的排列一共有8!=40320个,枚举量不会超过它。

而至于如何枚举,则需要编写递归程序实现枚举。
当把问题分成若干步骤并递归求解时,如果当前步骤没有合法选择,则函数将返回上一级递归调用,这种现象叫做回溯。真因为这个原因,递归枚举算法常被称作回溯法。

关于代码说明。
1.既然是逐行放置,则皇后肯定不会横向攻击,因此只需要检查是否纵向和斜向攻击即可。
2.vis数组的使用:vis数组的确切意义是什么?它表示已经放置的皇后占据了那些列、主对角线、副对角线。vis[0][i]代表占据列;vis[1][x+i]代表占据副对角线;vis[2][x-i+n]代表
占据主对角线上的点,但y-x可能为负数,所以需要加n存取。

3.一般的,如果在回溯法中修改了辅助的全局变量,则一定要既是把它们恢复原状(除非故意保留所做修改!)
 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 #define MAX 110
 5 using namespace std;
 6 int n;
 7 int sum;
 8 int ans[MAX];
 9 int vis[3][MAX];///vis[0]代表列,vis[1]代表副对角线,vis[2]代表主对角线
10 void DFS(int x)
11 {
12     int i;
13     if(x==n+1)
14     {
15         sum++;
16         if(sum<=3)
17         {
18             for(i=1; i<=n; i++)
19             {
20                 if(i==1)
21                 {
22                     printf("%d",ans[i]);
23                 }
24                 else
25                 {
26                     printf(" %d",ans[i]);
27                 }
28             }
29             printf("\n");
30             return ;
31         }
32     }
33     for(i=1; i<=n; i++)
34     {
35         if(!vis[0][i]&&!vis[1][x+i]&&!vis[2][x-i+n])
36         {
37             vis[0][i]=vis[1][x+i]=vis[2][x-i+n]=1;
38             ans[x]=i;
39             DFS(x+1);
40             vis[0][i]=vis[1][x+i]=vis[2][x-i+n]=0;///状态恢复
41         }
42     }
43 }
44 int main()
45 {
46     scanf("%d",&n);
47     memset(ans,0,sizeof(ans));
48     memset(vis,0,sizeof(vis));
49     DFS(1);
50     printf("%d\n",sum);
51 }

 





posted @ 2018-10-04 19:09  王陸  阅读(770)  评论(0编辑  收藏  举报