Leetcode 52. N-Queens II

回溯法替劈,非遞歸求 N 皇后問題解個數(shù)赔硫,Python 3 實現(xiàn):

源代碼已上傳 Github岩调,持續(xù)更新克饶。

"""
52. N-Queens II

Follow up for N-Queens problem.
Now, instead outputting board configurations, return the total number of distinct solutions.

"""


class Solution:

    # 判斷同一列和同一斜線上是否存在 Queue
    def is_valid(self, row, column, queen_record):
        c_left = column
        c_right = column
        for r in range(row, -1, -1):
            if queen_record[r] == column or queen_record[r] == c_left or queen_record[r] == c_right:
                return False
            c_left -= 1
            c_right += 1
        return True

    def totalNQueens(self, n):
        """
        :type n: int
        :rtype: int
        """
        chess_board = [['.' for x in range(n)] for y in range(n)]

        # 初始化為 - 1,表示當(dāng)前行還未放 Queue
        queen_record = [-1 for x in range(n)]
        solutions = []

        row = 0
        column = 0
        has_solution = True
        while has_solution:

            while row in range(n):

                while column < n or queen_record[row] == -1:

                    if column < n and self.is_valid(row, column, queen_record):
                        chess_board[row][column] = 'Q'
                        queen_record[row] = column
                        break
                    else:
                        column += 1

                        if column >= n:
                            # 當(dāng)前行無法放置 Queue 開始回溯
                            row = row - 1
                            if row < 0:
                                has_solution = False
                                break

                            chess_board[row][queen_record[row]] = '.'
                            column = queen_record[row] + 1
                            queen_record[row] = -1

                if has_solution:
                    column = 0
                    row += 1

            if has_solution:
                solutions.append(chess_board)

                # 回溯
                row -= 1
                chess_board[row][queen_record[row]] = '.'
                column = queen_record[row] + 1
                queen_record[row] = -1

        return len(solutions)


if __name__ == '__main__':
    solution = Solution()
    result = solution.totalNQueens(4)
    print(result)

源代碼已上傳至 Github誊辉,持續(xù)更新中矾湃。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市堕澄,隨后出現(xiàn)的幾起案子邀跃,更是在濱河造成了極大的恐慌,老刑警劉巖蛙紫,帶你破解...
    沈念sama閱讀 222,590評論 6 517
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件拍屑,死亡現(xiàn)場離奇詭異,居然都是意外死亡坑傅,警方通過查閱死者的電腦和手機僵驰,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,157評論 3 399
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來唁毒,“玉大人蒜茴,你說我怎么就攤上這事〗鳎” “怎么了粉私?”我有些...
    開封第一講書人閱讀 169,301評論 0 362
  • 文/不壞的土叔 我叫張陵,是天一觀的道長近零。 經(jīng)常有香客問我诺核,道長,這世上最難降的妖魔是什么久信? 我笑而不...
    開封第一講書人閱讀 60,078評論 1 300
  • 正文 為了忘掉前任窖杀,我火速辦了婚禮,結(jié)果婚禮上裙士,老公的妹妹穿的比我還像新娘入客。我一直安慰自己,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 69,082評論 6 398
  • 文/花漫 我一把揭開白布痊项。 她就那樣靜靜地躺著锅风,像睡著了一般酥诽。 火紅的嫁衣襯著肌膚如雪鞍泉。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 52,682評論 1 312
  • 那天肮帐,我揣著相機與錄音咖驮,去河邊找鬼。 笑死训枢,一個胖子當(dāng)著我的面吹牛托修,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播恒界,決...
    沈念sama閱讀 41,155評論 3 422
  • 文/蒼蘭香墨 我猛地睜開眼睦刃,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了十酣?” 一聲冷哼從身側(cè)響起涩拙,我...
    開封第一講書人閱讀 40,098評論 0 277
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎耸采,沒想到半個月后兴泥,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,638評論 1 319
  • 正文 獨居荒郊野嶺守林人離奇死亡虾宇,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,701評論 3 342
  • 正文 我和宋清朗相戀三年搓彻,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片嘱朽。...
    茶點故事閱讀 40,852評論 1 353
  • 序言:一個原本活蹦亂跳的男人離奇死亡旭贬,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出搪泳,到底是詐尸還是另有隱情骑篙,我是刑警寧澤,帶...
    沈念sama閱讀 36,520評論 5 351
  • 正文 年R本政府宣布森书,位于F島的核電站靶端,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏凛膏。R本人自食惡果不足惜杨名,卻給世界環(huán)境...
    茶點故事閱讀 42,181評論 3 335
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望猖毫。 院中可真熱鬧台谍,春花似錦、人聲如沸吁断。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,674評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至掷伙,卻和暖如春是己,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背任柜。 一陣腳步聲響...
    開封第一講書人閱讀 33,788評論 1 274
  • 我被黑心中介騙來泰國打工卒废, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人宙地。 一個月前我還...
    沈念sama閱讀 49,279評論 3 379
  • 正文 我出身青樓摔认,卻偏偏與公主長得像,于是被迫代替她去往敵國和親宅粥。 傳聞我的和親對象是個殘疾皇子参袱,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 45,851評論 2 361