// N Queens Problem
// 試探-回溯算法,遞歸實現
// sum用來記錄皇后放置成功的不同佈局數;upperlim用來標記所有列都已經放置好了皇后。
long sum = 0, upperlim = 1;
// 試探算法從最右邊的列開始。
void test(long row, long ld, long rd) 。
{
if (row != upperlim)
{
// row,ld,rd進行“或”運算,求得所有可以放置皇后的列,對應位爲0,
// 然後再取反後“與”上全1的數,來求得當前所有可以放置皇后的位置,對應列改爲1。
// 也就是求取當前哪些列可以放置皇后。
long pos = upperlim & ~(row | ld | rd);
while (pos) // 0 -- 皇后沒有地方可放,回溯。
{
// 拷貝pos最右邊爲1的bit,其餘bit置0。
// 也就是取得可以放皇后的最右邊的列。
long p = pos & -pos;
// 將pos最右邊爲1的bit清零。
// 也就是爲獲取下一次的最右可用列使用做準備,
// 程序將來會回溯到這個位置繼續試探。
pos -= p;
// row + p,將當前列置1,表示記錄這次皇后放置的列。
// (ld + p) << 1,標記當前皇后左邊相鄰的列不允許下一個皇后放置。
// (ld + p) >> 1,標記當前皇后右邊相鄰的列不允許下一個皇后放置。
// 此處的移位操作實際上是記錄對角線上的限制,只是因爲問題都化歸
// 到一行網格上來解決,所以表示爲列的限制就可以了。顯然,隨着移位
// 在每次選擇列之前進行,原來N×N網格中某個已放置的皇后針對其對角線
// 上產生的限制都被記錄下來了。
test(row + p, (ld + p) << 1, (rd + p) >> 1);
}
}
else
{
// row的所有位都爲1,即找到了一個成功的佈局,回溯。
sum++;
}
}
int main(int argc, char *argv[])
{
time_t tm;
int n = 16;
if (argc != 1)
n = atoi(argv[1]);
tm = time(0);
// 因爲整型數的限制,最大隻能32位,
// 如果想處理N大於32的皇后問題,需要
// 用bitset數據結構進行存儲。
if ((n < 1) || (n > 32))
{
printf(" 只能計算1-32之間/n");
exit(-1);
}
printf("%d 皇后/n", n);
// N個皇后只需N位存儲,N列中某列有皇后則對應bit置1。
upperlim = (upperlim << n) - 1;
test(0, 0, 0);
printf("共有%ld種排列, 計算時間%d秒 /n", sum, (int) (time(0) - tm));
}
上述代碼容易看懂,但我覺得核心的是在針對試探-回溯算法所用的數據結構的設計上。
程序採用了遞歸,也就是借用了編譯系統提供的自動回溯功能。
算法的核心:使用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來說仍能較大地提升效率!