算法【N皇后问题位运算实现】

avatar
作者
猴君
阅读量:0

N皇后问题是一个经典的回溯算法问题,解决N皇后问题的时间复杂度是O(n!),好的方法可以大量剪枝,大量优化常数时间。

用数组表示路径的方法(经典、常数时间慢,不推荐)

1.记录之前每一行的皇后放在了什么列

2.来到第i行的时候,可以根据0..i-1行皇后的位置,判断能放哪些列 

3.把能放的列都尝试一遍,每次尝试修改路径数组表示当前的决策,后续返回的答案都累加返回

用位运算的方法(巧妙、常数时间快,推荐)

1.int col: 0..i-1行皇后放置的位置因为正下方向延伸的原因,哪些列不能再放皇后

2.int left: 0..i-1行皇后放置的位置因为左下方向延伸的原因,哪些列不能再放皇后

3.int right: 0..i-1行皇后放置的位置因为右下方向延伸的原因,哪些列不能再放皇后

4.根据col、left、right,用位运算快速判断能放哪些列

5.把能放的列都尝试一遍,每次尝试修改3个数字表示当前的决策,后续返回的答案都累加返回

测试链接:https://leetcode.cn/problems/n-queens-ii/

分析:对于位置能否放置皇后采用位运算的方式判断。

代码如下。

class Solution { public:     int totalNQueens(int n) {         int ans = 0;         f(0, 0, 0, 0, ans, n);         return ans;     }      void f(int row, int col, int left, int right, int &ans, int n){         if(row == n){             ans++;             return;         }         int ban = ~(col | left | right);         for(int i = 0;i < n;++i){             if(((ban >> i) & 1) == 1){                 int place = (1 << i);                 f(row+1, col | place, (left | place)>>1, (right | place)<<1, ans, n);             }         }     } };

其中,ans表示个数,row表示进行到第几行,col、left、right表示分别从竖直、右上到左下、左上到右下三个方向上哪些列不能被放置,1为不能,0为能。ban在col、left、right或之后取反可以得到哪些列还可以放置,1为可放置,0为不可放置。遍历找到可放置的列,place中1的位数代表当前行皇后放在哪列。更新col、left、right。 

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!