Jednoduchý backtrackingový algoritmus pro řešení sudoku. Tento script prohledává všechny možné kombinace čísel v mřížce a vrací řešení, pokud existuje.
Ve skriptu stačí nadefinovat vstupní rozložení čísel na mřížce (0 představuje prázdné pole) a spustit funkci solve(), která vypíše všechny možné řešení pro danou mřížku.
Ve scriptu je nastaveno toto Sudoku pole:
Script najde řešení a vypíše ho na obrazovku, stiskem klávesy enter provedete novou analýzu, pokud bude existovat další řešení, opět se vypíše, a to tak dlouho, dokud nebudou vyčerpány všechny možnosti.
import numpy as np
grid = [[5,3,0,0,7,0,0,0,0],
[6,0,0,1,9,5,0,0,0],
[0,9,8,0,0,0,0,6,0],
[8,0,0,0,6,0,0,0,3],
[4,0,0,8,0,3,0,0,1],
[7,0,0,0,2,0,0,0,6],
[0,6,0,0,0,0,0,8,0],
[0,0,0,4,1,9,0,0,5],
[0,0,0,0,8,0,0,7,9]]
def possible(y,x,n):
global grid
for i in range(0,9):
if grid[y][i] == n:
return False
for i in range(0,9):
if grid[i][x] == n:
return False
x0 = (x//3)*3
y0 = (y//3)*3
for i in range(0,3):
for j in range(0,3):
if grid[y0+i][x0+j] == n:
return False
return True
def solve():
global grid
for y in range(9):
for x in range(9):
if grid[y][x] == 0:
for n in range(1,10):
if possible(y,x,n):
grid[y][x] = n
solve()
grid[y][x] = 0
return
print(np.matrix(grid))
print("\n")
input("Další možné řešení? (stiskni Enter)")
print("\n")
solve()