核心代碼如下:
1 void test(int row, int ld, int rd) 2 { 3 int pos, p; 4 if ( row != upperlim ) 5 { 6 pos = upperlim & (~(row | ld | rd )); 7 while ( pos ) 8 { 9 p = pos & (~pos + 1); 10 pos = pos - p; 11 test(row | p, (ld | p) << 1, (rd | p) >> 1); 12 } 13 } 14 else 15 ++Ans; 16 }
初始化: upperlim = (1 << n)-1; Ans = 0;
調用參數:test(0, 0, 0);
和普通算法一樣,這是一個遞歸函數,程序一行一行地尋找可以放皇后的地方。函數帶三個參數row、ld和rd,分別表示在縱列和兩個對角線方向的限制條件下這一行的哪些地方不能放。位於該行上的沖突位置就用row、ld和rd中的1來表示。把它們三個並起來,得到該行所有的禁位,取反后就得到所有可以放的位置(用pos來表示)。
p = pos & (~pos + 1)其結果是取出最右邊的那個1。這樣,p就表示該行的某個可以放子的位置,把它從pos中移除並遞歸調用test過程。
注意遞歸調用時三個參數的變化,每個參數都加上了一個禁位,但兩個對角線方向的禁位對下一行的影響需要平移一位。最后,如果遞歸到某個時候發現row=upperlim了,說明n個皇后全放進去了,找到的解的個數加一。
注:
upperlime:=(1 << n)-1 就生成了n個1組成的二進制數。
這個程序是從上向下搜索的。
pos & -pos 的意思就是取最右邊的 1 再組成二進制數,相當於 pos &(~pos +1),因為取反以后剛好所有數都是相反的(怎么聽着像廢話),再加 1 ,就是改變最低位,如果低位的幾個數都是1,加的這個 1 就會進上去,一直進到 0 ,在做與運算就和原數對應的 1 重合了。舉例可以說明:
原數 0 0 0 0 1 0 0 0 原數 0 1 0 1 0 0 1 1
取反 1 1 1 1 0 1 1 1 取反 1 0 1 0 1 1 0 0
加1 1 1 1 1 1 0 0 0 加1 1 0 1 0 1 1 0 1
與運算 0 0 0 0 1 0 0 0 and 0 0 0 0 0 0 0 1
其中呢,這個取反再加 1 就是補碼,and 運算 與負數,就是按位和補碼與運算。
(ld | p)<< 1 是因為由ld造成的占位在下一行要右移一下;
(rd | p)>> 1 是因為由rd造成的占位在下一行要左移一下。
ld rd row 還要和upperlime 與運算 一下,這樣做的結果就是從最低位數起取n個數為有效位置,原因是在上一次的運算中ld發生了右移,如果不and的話,就會誤把n以外的位置當做有效位。
pos 已經完成任務了還要減去p 是因為?
while 循環是因為?
在進行到某一層的搜索時,pos中存儲了所有的可放位置,為了求出所有解,必須遍歷所有可放的位置,而每走過一個點必須要刪掉它,否則就成死循環啦!
這個是目前公認N皇后的最高效算法。
完整的代碼如下:
1 /* 2 ** 目前最快的N皇后遞歸解決方法 3 ** N Queens Problem 4 ** 試探-回溯算法,遞歸實現 5 */ 6 #include "iostream" 7 using namespace std; 8 #include "time.h" 9 10 // sum用來記錄皇后放置成功的不同布局數;upperlim用來標記所有列都已經放置好了皇后。 11 long sum = 0, upperlim = 1; 12 13 // 試探算法從最右邊的列開始。 14 void test(long row, long ld, long rd) 15 { 16 if (row != upperlim) 17 { 18 // row,ld,rd進行“或”運算,求得所有可以放置皇后的列,對應位為0, 19 // 然后再取反后“與”上全1的數,來求得當前所有可以放置皇后的位置,對應列改為1 20 // 也就是求取當前哪些列可以放置皇后 21 long pos = upperlim & ~(row | ld | rd); 22 while (pos) // 0 -- 皇后沒有地方可放,回溯 23 { 24 // 拷貝pos最右邊為1的bit,其余bit置0 25 // 也就是取得可以放皇后的最右邊的列 26 long p = pos & -pos; 27 28 // 將pos最右邊為1的bit清零 29 // 也就是為獲取下一次的最右可用列使用做准備, 30 // 程序將來會回溯到這個位置繼續試探 31 pos -= p; 32 33 // row + p,將當前列置1,表示記錄這次皇后放置的列。 34 // (ld + p) << 1,標記當前皇后左邊相鄰的列不允許下一個皇后放置。 35 // (ld + p) >> 1,標記當前皇后右邊相鄰的列不允許下一個皇后放置。 36 // 此處的移位操作實際上是記錄對角線上的限制,只是因為問題都化歸 37 // 到一行網格上來解決,所以表示為列的限制就可以了。顯然,隨着移位 38 // 在每次選擇列之前進行,原來N×N網格中某個已放置的皇后針對其對角線 39 // 上產生的限制都被記錄下來了 40 test(row + p, (ld + p) << 1, (rd + p) >> 1); 41 } 42 } 43 else 44 { 45 // row的所有位都為1,即找到了一個成功的布局,回溯 46 sum++; 47 } 48 } 49 50 int main(int argc, char *argv[]) 51 { 52 time_t tm; 53 int n = 16; 54 55 if (argc != 1) 56 n = atoi(argv[1]); 57 tm = time(0); 58 59 // 因為整型數的限制,最大只能32位, 60 // 如果想處理N大於32的皇后問題,需要 61 // 用bitset數據結構進行存儲 62 if ((n < 1) || (n > 32)) 63 { 64 printf(" 只能計算1-32之間\n"); 65 exit(-1); 66 } 67 printf("%d 皇后\n", n); 68 69 // N個皇后只需N位存儲,N列中某列有皇后則對應bit置1。 70 upperlim = (upperlim << n) - 1; 71 72 test(0, 0, 0); 73 printf("共有%ld種排列, 計算時間%d秒 \n", sum, (int) (time(0) - tm)); 74 system("pause"); 75 return 0; 76 }
上述代碼還是比較容易看懂的,但我覺得核心的是在針對試探-回溯算法所用的數據結構的設計上。 程序采用了遞歸,也就是借用了編譯系統提供的自動回溯功能。
算法的核心:使用bit數組來代替以前由int或者bool數組來存儲當前格子被占用或者說可用信息,從這可以看出N個皇后對應需要N位表示。 巧妙之處在於:以前我們需要在一個N*N正方形的網格中挪動皇后來進行試探回溯,每走一步都要觀察和記錄一個格子前后左右對角線上格子的信息;采用bit位進行信息存儲的話,就可以只在一行格子也就是(1行×N列)個格子中進行試探回溯即可,對角線上的限制被化歸為列上的限制。
程序中主要需要下面三個bit數組,每位對應網格的一列,在C中就是取一個整形數的某部分連續位即可。 row用來記錄當前哪些列上的位置不可用,也就是哪些列被皇后占用,對應為1。ld,rd同樣也是記錄當前哪些列位置不可用,但是不表示被皇后占用,而是表示會被已有皇后在對角線上吃掉的位置。這三個位數組進行“或”操作后就是表示當前還有哪些位置可以放置新的皇后,對應0的位置可放新的皇后。如下圖所示的8皇后問題求解得第一步:
row: [ ][ ][ ][ ][ ][ ][ ][*]
ld: [ ][ ][ ][ ][ ][ ][*][ ]
rd: [ ][ ][ ][ ][ ][ ][ ][ ]
--------------------------------------
row|ld|rd: [ ][ ][ ][ ][ ][ ][*][*]
所有下一個位置的試探過程都是通過位操作來實現的,這是借用了C語言的好處,詳見代碼注釋。
關於此算法,如果考慮N×N棋盤的對稱性,對於大N來說仍能較大地提升效率! 位操作--對優化算法有了個新的認識
這個是在csdn找到的一個N皇后問題最快的算法,看了好一會才明白,這算法巧妙之處我認為有2個:
1、以前都是用數組來描述狀態,而這算法采用是的位來描述,運算速度可以大大提升,以后寫程序對於描述狀態的變量大家可以借鑒這個例子,會讓你的程序跑得更快
2、描述每行可放置的位置都是只用row,ld,rd這3個變量來描述,這樣使得程序看起來挺簡潔的。
本文轉自:http://blog.csdn.net/kai_wei_zhang/article/details/8033194