daniel/kenken
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

allow grid to be between 2 and 9

Browse Source
This commit is contained in:
Daniel Tsvetkov 2022-04-09 11:19:56 +02:00
parent 9997434ac2
commit 47ccffaf81
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
main.py
View File

@ -24,12 +24,12 @@ OP_LAMBDAS = {
OP_NONE: lambda x: x[0], OP_NONE: lambda x: x[0],
} }
BOX_H_LABELS = 'ABCDEF' BOX_H_LABELS = 'ABCDEFGHI'
BOX_V_LABELS = '654321' BOX_V_LABELS = '123456789'
def translate_box_to_rc(box): def translate_box_to_rc(box, grid_size):
return BOX_V_LABELS.index(box[1]), BOX_H_LABELS.index(box[0]) return BOX_V_LABELS.index(box[1]), BOX_H_LABELS[grid_size-1::-1].index(box[0])
def has_line_integrity(line): def has_line_integrity(line):
@ -47,7 +47,7 @@ def fill_grid(grid, solution):
for box_value in solution: for box_value in solution:
box_values.append(box_value) box_values.append(box_value)
for box, value in box_values: for box, value in box_values:
br, bc = translate_box_to_rc(box) br, bc = translate_box_to_rc(box, len(grid))
grid[br][bc] = value grid[br][bc] = value
return grid return grid
@ -145,7 +145,8 @@ class Block(object):
class Game(object): class Game(object):
def __init__(self, grid_size, blocks): def __init__(self, grid_size: int, blocks):
assert 2 <= grid_size <= 9
self.blocks = blocks self.blocks = blocks
self.grid_size = grid_size self.grid_size = grid_size
self.verify() self.verify()