/
sudoku.py
118 lines (88 loc) · 2.3 KB
/
sudoku.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
import copy
board = [
".........",
".........",
".........",
".........",
".........",
".........",
".........",
".........",
"........."
]
def main():
global board
for idx, line in enumerate(board):
board[idx] = list(line)
solve()
printBoard()
def solve():
global board
try:
fillAllObvious()
except:
return False
if isComplete():
return True
i, j = 0, 0
for rowIdx, row in enumerate(board):
for colIdx, col in enumerate(row):
if col == ".":
i, j = rowIdx, colIdx
possibilities = getPossibilities(i, j)
for value in possibilities:
snapshot = copy.deepcopy(board)
board[i][j] = value
result = solve()
if result == True:
return True
else:
board = copy.deepcopy(snapshot)
return False
def fillAllObvious():
global board
while True:
somethingChanged = False
for i in range(0, 9):
for j in range(0, 9):
possibilities = getPossibilities(i, j)
if possibilities == False:
continue
if len(possibilities) == 0:
raise RuntimeError("No moves left")
if len(possibilities) == 1:
board[i][j] = possibilities[0]
somethingChanged = True
if somethingChanged == False:
return
def getPossibilities(i, j):
global board
if board[i][j] != ".":
return False
possibilities = {str(n) for n in range(1, 10)}
for val in board[i]:
possibilities -= set(val)
for idx in range(0, 9):
possibilities -= set(board[idx][j])
iStart = (i // 3) * 3
jStart = (j // 3) * 3
subboard = board[iStart:iStart + 3]
for idx, row in enumerate(subboard):
subboard[idx] = row[jStart:jStart + 3]
for row in subboard:
for col in row:
possibilities -= set(col)
return list(possibilities)
def printBoard():
global board
for row in board:
for col in row:
print(col, end="")
print("")
def isComplete():
for row in board:
for col in row:
if (col == "."):
return False
return True
main()