Skip to content

Instantly share code, notes, and snippets.

@syphh
Created April 3, 2022 14:13
Show Gist options
  • Star 11 You must be signed in to star a gist
  • Fork 4 You must be signed in to fork a gist
  • Save syphh/62e6140361feb2d7196f2cb050c987b3 to your computer and use it in GitHub Desktop.
Save syphh/62e6140361feb2d7196f2cb050c987b3 to your computer and use it in GitHub Desktop.
def is_valid(grid, r, c, k):
not_in_row = k not in grid[r]
not_in_column = k not in [grid[i][c] for i in range(9)]
not_in_box = k not in [grid[i][j] for i in range(r//3*3, r//3*3+3) for j in range(c//3*3, c//3*3+3)]
return not_in_row and not_in_column and not_in_box
def solve(grid, r=0, c=0):
if r == 9:
return True
elif c == 9:
return solve(grid, r+1, 0)
elif grid[r][c] != 0:
return solve(grid, r, c+1)
else:
for k in range(1, 10):
if is_valid(grid, r, c, k):
grid[r][c] = k
if solve(grid, r, c+1):
return True
grid[r][c] = 0
return False
grid = [
[0, 0, 4, 0, 5, 0, 0, 0, 0],
[9, 0, 0, 7, 3, 4, 6, 0, 0],
[0, 0, 3, 0, 2, 1, 0, 4, 9],
[0, 3, 5, 0, 9, 0, 4, 8, 0],
[0, 9, 0, 0, 0, 0, 0, 3, 0],
[0, 7, 6, 0, 1, 0, 9, 2, 0],
[3, 1, 0, 9, 7, 0, 2, 0, 0],
[0, 0, 9, 1, 8, 2, 0, 0, 3],
[0, 0, 0, 0, 6, 0, 1, 0, 0]
]
solve(grid)
print(*grid, sep='\n')
@Fiedler01
Copy link

Fiedler01 commented Jan 7, 2024

This solution is nice and clear. Turning the "solve" function into a generator for ALL solutions of the SuDoKu is straightforward.

def solve(grid, r=0, c=0):
    if r == 9:
        yield [row[:] for row in grid]  # yield a copy of the grid as a solution
    elif c == 9:
        yield from solve(grid, r + 1, 0)
    elif grid[r][c] != 0:
        yield from solve(grid, r, c + 1)
    else:
        for k in range(1, 10):
            if is_valid(grid, r, c, k):
                grid[r][c] = k
                yield from solve(grid, r, c + 1)
                grid[r][c] = 0

for solution in solve([row[:] for row in grid]):
    print(*solution, sep='\n')
    print('-' * 20)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment