背景

在LintCode刷题的时候碰到一个很水的题目,不过要注意很多细节,不然调试的时候非常麻烦,利用Python的一些小技巧写了一个很简短的解决方案。

题目描述

请判定一个数独是否有效。

该数独可能只填充了部分数字,其中缺少的数字用 . 表示。

注意事项

一个合法的数独(仅部分填充)并不一定是可解的。我们仅需使填充的空格有效即可。

解体思路

  1. 将数独按照行、列和块进行预处理,然后分别判断是否合法。

  2. 利用Python的表达式推导,匿名函数和all函数可以很方便的进行处理。

代码

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]) # 一步搞定
# 感想
能不用循环体尽量不用循环体,她好,我也好。

忆先
4.2k 声望45 粉丝