经典算法系列之n皇后问题

N皇后问题是一个经典的问题，在一个N*N的棋盘上放置N个皇后，每行一个并使其不能互相攻击（同一行、同一列、同一斜线上的皇后都会自动攻击）。

求解N皇后问题是算法中回溯法应用的一个经典案例

回溯算法也叫试探法，它是一种系统地搜索问题的解的方法。回溯算法的基本思想是：从一条路往前走，能进则进，不能进则退回来，换一条路再试。

      在现实中，有很多问题往往需要我们把其所有可能穷举出来，然后从中找出满足某种要求的可能或最优的情况，从而得到整个问题的解。回溯算法就是解决这种问题的“通用算法”，有“万能算法”之称。N皇后问题在N增大时就是这样一个解空间很大的问题，所以比较适合用这种方法求解。这也是N皇后问题的传统解法，很经典。

      下面是算法的高级伪码描述，这里用一个N*N的矩阵来存储棋盘：

      1) 算法开始, 清空棋盘，当前行设为第一行，当前列设为第一列

      2) 在当前行，当前列的位置上判断是否满足条件(即保证经过这一点的行,列与斜线上都没有两个皇后)，若不满足，跳到第4步

      3) 在当前位置上满足条件的情形：

                 在当前位置放一个皇后，若当前行是最后一行，记录一个解；

                 若当前行不是最后一行，当前行设为下一行, 当前列设为当前行的第一个待测位置；

                 若当前行是最后一行，当前列不是最后一列，当前列设为下一列；

                 若当前行是最后一行，当前列是最后一列，回溯，即清空当前行及以下各行的棋盘，然后，当前行设为上一行，当前列设为当前行的下一个待测位置；

                以上返回到第2步

      4) 在当前位置上不满足条件的情形：

                若当前列不是最后一列，当前列设为下一列，返回到第2步;

                若当前列是最后一列了，回溯，即，若当前行已经是第一行了，算法退出，否则，清空当前行及以下各行的棋盘，然后，当前行设为上一行，当前列设为当前行的下一个待测位置，返回到第2步; 

        算法的基本原理是上面这个样子，但不同的是用的数据结构不同，检查某个位置是否满足条件的方法也不同。为了提高效率，有各种优化策略，如多线程，多分配内存表示棋盘等。

        在具体解决该问题时，可以将其拆分为几个小问题。首先就是在棋盘上如何判断两个皇后是否能够相互攻击，在最初接触这个问题时，首先想到的方法就是把棋盘存储为一个二维数组，然后在需要在第i行第j列放置皇后时，根据问题的描述，首先判断是在第i行是否有皇后，由于每行只有一个皇后，这个判断也可以省略，然后判断第j列是否有皇后，这个也很简单，最后需要判断在同一斜线上是否有皇后，按照该方法需要判断两次，正对角线方向和负对角线方向，总体来说也不难。但是写完之后，总感觉很笨，因为在N皇后问题中这个函数的使用次数太多了，而这样做效率较差，个人感觉很不爽。上网查看了别人的实现之后大吃一惊，大牛们都是使用一个一维数组来存储棋盘，在某个位置上是否有皇后可以相互攻击的判断也很简单。具体细节如下：

        把棋盘存储为一个N维数组a[N]，数组中第i个元素的值代表第i行的皇后位置，这样便可以把问题的空间规模压缩为一维O(N)，在判断是否冲突时也很简单，首先每行只有一个皇后，且在数组中只占据一个元素的位置，行冲突就不存在了，其次是列冲突，判断一下是否有a[i]与当前要放置皇后的列j相等即可。至于斜线冲突，通过观察可以发现所有在斜线上冲突的皇后的位置都有规律即它们所在的行列互减的绝对值相等，即|  row – i | = | col – a[i] | 。这样某个位置是否可以放置皇后的问题已经解决。

       下面要解决的是使用何种方法来找到所有的N皇后的解。上面说过该问题是回溯法的经典应用，所以可以使用回溯法来解决该问题，具体实现也有两个途径，递归和非递归。递归方法较为简单，大致思想如下：

     void queen(int row)

    {

              if (n == row)      //如果已经找到结果，则打印结果

                    print_result();

              else {

                          for (k=0 to N) { //试探第row行每一个列

                                  if (can_place(row, k) { 

                                          place(row, k);   //放置皇后

                                         queen(row + 1);  //继续探测下一行

                                  }

                         }

             }

    }

        该方法由于在探测第i行后，如果找到一个可以放置皇后的位置j后，则会递归探测下一行，结束后则会继续探测i行j+1列，故可以找到所有的N皇后的解。

        但是一般来说递归的效率比较差，下面重点讨论一下该问题的非递归实现。

        非递归方法的一个重要问题时何时回溯及如何回溯的问题。程序首先对N行中的每一行进行探测，寻找该行中可以放置皇后的位置，具体方法是对该行的每一列进行探测，看是否可以放置皇后，如果可以，则在该列放置一个皇后，然后继续探测下一行的皇后位置。如果已经探测完所有的列都没有找到可以放置皇后的列，此时就应该回溯，把上一行皇后的位置往后移一列，如果上一行皇后移动后也找不到位置，则继续回溯直至某一行找到皇后的位置或回溯到第一行，如果第一行皇后也无法找到可以放置皇后的位置，则说明已经找到所有的解程序终止。如果该行已经是最后一行，则探测完该行后，如果找到放置皇后的位置，则说明找到一个结果，打印出来。但是此时并不能再此处结束程序，因为我们要找的是所有N皇后问题所有的解，此时应该清除该行的皇后，从当前放置皇后列数的下一列继续探测。

完整的代码如下：

/*** 回溯法解N皇后问题* 使用一个一维数组表示皇后的位置* 其中数组的下标表示皇后所在的行* 数组元素的值表示皇后所在的列* 这样设计的棋盘，所有皇后必定不在同一行，于是行冲突就不存在了**/#include <stdio.h>#include <stdlib.h>#include <math.h>#define QUEEN 8     //皇后的数目#define INITIAL -10000   //棋盘的初始值int a[QUEEN];    //一维数组表示棋盘void init()  //对棋盘进行初始化{int *p;for (p = a; p < a + QUEEN; ++p) {*p = INITIAL;}} int valid(int row, int col)    //判断第row行第col列是否可以放置皇后{int i;for (i = 0; i < QUEEN; ++i)   //对棋盘进行扫描{if (a[i] == col || abs(i - row) == abs(a[i] - col))   //判断列冲突与斜线上的冲突return 0;}return 1;} void print()    //打印输出N皇后的一组解{int i, j;for (i = 0; i < QUEEN; ++i){for (j = 0; j < QUEEN; ++j){if (a[i] != j)      //a[i]为初始值printf("%c ", '.');else                //a[i]表示在第i行的第a[i]列可以放置皇后printf("%c ", '#');}printf("\n");}for (i = 0; i < QUEEN; ++i)printf("%d ", a[i]);printf("\n");printf("--------------------------------\n");}void queen()      //N皇后程序{int n = 0;int i = 0, j = 0;while (i < QUEEN){while (j < QUEEN)        //对i行的每一列进行探测，看是否可以放置皇后{if(valid(i, j))      //该位置可以放置皇后{a[i] = j;        //第i行放置皇后j = 0;           //第i行放置皇后以后，需要继续探测下一行的皇后位置，所以此处将j清零，从下一行的第0列开始逐列探测break;}else{++j;             //继续探测下一列}}if(a[i] == INITIAL)         //第i行没有找到可以放置皇后的位置{if (i == 0)             //回溯到第一行，仍然无法找到可以放置皇后的位置，则说明已经找到所有的解，程序终止break;else                    //没有找到可以放置皇后的列，此时就应该回溯{--i;j = a[i] + 1;        //把上一行皇后的位置往后移一列a[i] = INITIAL;      //把上一行皇后的位置清除，重新探测continue;}}if (i == QUEEN - 1)          //最后一行找到了一个皇后位置，说明找到一个结果，打印出来{printf("answer %d : \n", ++n);print();//不能在此处结束程序，因为我们要找的是N皇后问题的所有解，此时应该清除该行的皇后，从当前放置皇后列数的下一列继续探测。//_sleep(600);j = a[i] + 1;             //从最后一行放置皇后列数的下一列继续探测a[i] = INITIAL;           //清除最后一行的皇后位置continue;}++i;              //继续探测下一行的皇后位置}}int main(void){init();queen();system("pause");return 0;}