进入下一行时,col 的值保持不变,pie左移一位,na右移一位,由于棋盘的最左列对应每个整数的最低二进制位,即每个整数的最右二进制位,因此对整数的移位操作方向和对棋盘的移位操作方向相反(对棋盘的移位操作方向是pei右移一位,na左移一位)
class Solution:
def totalNQueens(self, n: int) -> int:
self.count = 0
self.n = n
self.dfs(0, 0, 0, 0)
return self.count
# 注意函数本身并没有任何返回值,返回的的只是对象本身的变量
def dfs(self, row, col, pie, na):
if row >= self.n:
self.count += 1
return self.count
# 当前可以放置的地方, 结果为 1的位是可以放置的位
avaliableBits = (~(col | pie | na)) & ((1<<self.n) - 1)
while(avaliableBits > 0) :
p = avaliableBits & (-avaliableBits)
self.dfs(row + 1, (col | p), (pie | p) << 1, (na | p) >> 1)
avaliableBits &= (avaliableBits - 1)