【Python】判斷數獨是否合法

背景

在LintCode刷題的時候碰到一個很水的題目,不過要注意不少細節,否則調試的時候很是麻煩,利用Python的一些小技巧寫了一個很簡短的解決方案。函數

題目描述

請斷定一個數獨是否有效。調試

該數獨可能只填充了部分數字,其中缺乏的數字用 . 表示。code

注意事項io

一個合法的數獨(僅部分填充)並不必定是可解的。咱們僅需使填充的空格有效便可。class

解體思路

  1. 將數獨按照行、列和塊進行預處理,而後分別判斷是否合法。匿名函數

  2. 利用Python的表達式推導,匿名函數和all函數能夠很方便的進行處理。lambda

代碼

class Solution:循環

# @param board, a 9x9 2D array
# @return a boolean
def isValidSudoku(self, board):
    rows = [list(lst[::]) for lst in board]
    columns = [[lst[idx] for lst in board] for idx in range(9)]
    blocks_origin = [board[row][column] for x in [[0, 1, 2], [3, 4, 5], [6, 7, 8]] for y in [[0, 1, 2], [3, 4, 5], [6, 7, 8]] for row in x for column in y] # 利用一些技巧直接按塊儲存數據
    blocks = [[blocks_origin[row * 9 + column] for column in range(9)] for row in range(9)]
    check = lambda lst: all([lst.count(x) == 1 for x in lst if x != '.']) # 判斷一條記錄(按某種方式排列的九個數字)是否合法
    return all([check(x) for style in (rows, columns, blocks) for x in style]) # 一步搞定
# 感想
能不用循環體儘可能不用循環體,她好,我也好。
相關文章
相關標籤/搜索