python八皇后問題的解決方法

class EightQueensPuzzle(object):
  '''
  八皇后問題求解
  代碼使用方法：
    eight_q = EightQueensPuzzle(4, 5)
    print "EIGHT QUEEDS PUZZLE:"
    result = eight_q.eight_queens_puzzle()
    for i in result:
      print i
  '''
  def __init__(self, n, char):
    self.n = n # 棋盤維度
    self.char = char # 皇后標(biāo)記字符

  def init_chess_board(self, n):
    '''
    初始化一個(gè)棋盤，棋盤規(guī)格可以按參數(shù) n 隨意選定,一般都討論八皇后，就選擇 8
    :return: 返回棋盤，是一個(gè) 8*8 矩陣
    '''
    chess_board = []
    for i in xrange(0, n):
      line = []
      for j in xrange(0, n):
        line.append(0)
      chess_board.append(line)
    return chess_board

  def update_conflict_board(self, conflict_board, position):
    for k in xrange(0, self.n): # 為行添加 1
      conflict_board[position[0]][k] = 1
    for id in xrange(position[0]+1, self.n):
      conflict_board[id][position[1]] = 1 # 為列添加 1
      if position[0] + position[1] - id >= 0: # 為左斜添加 1
        conflict_board[id][position[0] + position[1] - id] = 1
      if position[1] - position[0] + id < self.n: # 為右斜添加 1
        conflict_board[id][position[1] - position[0] + id] = 1

  def queens_conflict(self, conflict_board, position):
    '''
    當(dāng)前棋盤的狀態(tài)是 conflict_board, 判定如果在 position 位置給一個(gè)皇后的話，會(huì)不會(huì)出現(xiàn)問題。
    如果有問題則返回 False，如果沒有問題返回 True
    '''
    if conflict_board[position[0]][position[1]] != 0:
      return False
    else:
      return True

  def eight_queens_puzzle(self):
    '''
    給出一個(gè)八皇后的求解答案。
    :return:返回一個(gè)結(jié)果并打印.
    '''
    import random
    while True: # 不停尋找符合條件的八皇后排列
      chess_board = self.init_chess_board(self.n)
      conflict_board = self.init_chess_board(self.n)
      for i in xrange(0, self.n):
        flag = 0
        for cnt in conflict_board[i]:
          if cnt != 0:
            flag += 1
        if flag == self.n: # 如果已經(jīng)1被填滿了，說明這個(gè)答案錯(cuò)誤
          break

        while True:
          pos = [i, random.randint(0, self.n-1)] # 元組構(gòu)成皇后的位置
          if self.queens_conflict(conflict_board, pos): # 如果沒有沖突
            chess_board[i][pos[1]] = self.char
            self.update_conflict_board(conflict_board, pos)
            break
      if self.char in chess_board[self.n-1]:
        return chess_board