Leetcode#51 N-Queens

原题地址

 

经典的8皇后问题变形版

当然也是用经典的8皇后问题解法的变形版咯(仿照"位运算求解8皇后问题")

row:当前是第几行

col:皇后所在的列的情况

left:左上角禁位(因为我是从下向上迭代的)

right:右上角禁位(因为我是从下向上迭代的)

 

代码:

 1 vector<vector<string> > result;
 2 vector<string> buffer;
 3 
 4 void solve(int row, vector<bool> &col, vector<bool> left, vector<bool> right) {
 5   if (row == 0) {
 6     result.push_back(buffer);
 7     return;
 8   }
 9 
10   // 计算当前行可以放置皇后的位
11   vector<bool> avail(col.size(), false);
12   for (int i = 0; i < col.size(); i++)
13     avail[i] = !(col[i] || left[i] || right[i]);
14 
15   // 计算上一行的左右禁位
16   for (int i = 0; i < col.size() - 1; i++)
17     left[i] = left[i + 1];
18   left[col.size() - 1] = false;
19   for (int i = col.size() - 1; i > 0; i--)
20     right[i] = right[i - 1];
21   right[0] = false;
22 
23   // 尝试、回溯
24   for (int i = 0; i < col.size(); i++) {
25     if (!avail[i])
26       continue;
27     col[i] = true;
28     if (i > 0) left[i - 1] = true;
29     if (i < col.size() - 1) right[i + 1] = true;
30     buffer.push_back(string(i, '.') + "Q" + string(col.size() - i - 1, '.'));
31     solve(row - 1, col, left, right);
32     buffer.pop_back();
33     if (i > 0) left[i - 1] = false;
34     if (i < col.size() - 1) right[i + 1] = false;
35     col[i] = false;
36   }
37 }

 

posted @ 2015-01-23 20:17  李舜阳  阅读(209)  评论(0编辑  收藏  举报