Added leetcode exercise 36
This commit is contained in:
parent
fb0684dd27
commit
3f5146d5a8
50
leetcode/36_valid_sudoku.py
Normal file
50
leetcode/36_valid_sudoku.py
Normal file
@ -0,0 +1,50 @@
|
||||
from typing import List
|
||||
import pytest
|
||||
|
||||
|
||||
class Solution:
|
||||
def isValidSudoku(self, board: List[List[str]]) -> bool:
|
||||
rows, cols, cubes = set(), set(), set()
|
||||
for i in range(9):
|
||||
for j in range(9):
|
||||
val = board[i][j]
|
||||
if not val.isdigit():
|
||||
continue
|
||||
|
||||
cube = (i // 3) * 3 + (j // 3)
|
||||
if (i, val) in rows or (j, val) in cols or (cube, val) in cubes:
|
||||
return False
|
||||
|
||||
rows.add((i, val))
|
||||
cols.add((j, val))
|
||||
cubes.add((cube, val))
|
||||
|
||||
return True
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def solution():
|
||||
return Solution()
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"sudoku, expected",
|
||||
[
|
||||
(
|
||||
[
|
||||
["5", "3", ".", ".", "7", ".", ".", ".", "."],
|
||||
["6", ".", ".", "1", "9", "5", ".", ".", "."],
|
||||
[".", "9", "8", ".", ".", ".", ".", "6", "."],
|
||||
["8", ".", ".", ".", "6", ".", ".", ".", "3"],
|
||||
["4", ".", ".", "8", ".", "3", ".", ".", "1"],
|
||||
["7", ".", ".", ".", "2", ".", ".", ".", "6"],
|
||||
[".", "6", ".", ".", ".", ".", "2", "8", "."],
|
||||
[".", ".", ".", "4", "1", "9", ".", ".", "5"],
|
||||
[".", ".", ".", ".", "8", ".", ".", "7", "9"],
|
||||
],
|
||||
True,
|
||||
)
|
||||
],
|
||||
)
|
||||
def test_search(solution: Solution, sudoku: List[List[str]], expected: bool):
|
||||
assert solution.isValidSudoku(sudoku) == expected
|
Loading…
x
Reference in New Issue
Block a user