1
1
"""
2
-
3
2
Given a partially filled 9×9 2D array, the objective is to fill a 9×9
4
3
square grid with digits numbered 1 to 9, so that every row, column, and
5
4
and each of the nine 3×3 sub-grids contains all of the digits.
9
8
function on the next column to see if it returns True. if yes, we
10
9
have solved the puzzle. else, we backtrack and place another number
11
10
in that cell and repeat this process.
12
-
13
11
"""
14
-
15
12
# assigning initial values to the grid
16
13
initial_grid = [
17
14
[3 , 0 , 6 , 5 , 0 , 8 , 4 , 0 , 0 ],
24
21
[0 , 0 , 0 , 0 , 0 , 0 , 0 , 7 , 4 ],
25
22
[0 , 0 , 5 , 2 , 0 , 6 , 3 , 0 , 0 ],
26
23
]
24
+
27
25
# a grid with no solution
28
26
no_solution = [
29
27
[5 , 0 , 6 , 5 , 0 , 8 , 4 , 0 , 3 ],
@@ -44,9 +42,7 @@ def is_safe(grid, row, column, n):
44
42
column, and the 3x3 subgrids contain the digit 'n'.
45
43
It returns False if it is not 'safe' (a duplicate digit
46
44
is found) else returns True if it is 'safe'
47
-
48
45
"""
49
-
50
46
for i in range (9 ):
51
47
if grid [row ][i ] == n or grid [i ][column ] == n :
52
48
return False
@@ -64,24 +60,15 @@ def is_completed(grid):
64
60
This function checks if the puzzle is completed or not.
65
61
it is completed when all the cells are assigned with a number(not zero)
66
62
and There is no repeating number in any column, row or 3x3 subgrid.
67
-
68
63
"""
69
-
70
- for row in grid :
71
- for cell in row :
72
- if cell == 0 :
73
- return False
74
-
75
- return True
64
+ return all (cell != 0 for cell in row for row in grid )
76
65
77
66
78
67
def find_empty_location (grid ):
79
68
"""
80
69
This function finds an empty location so that we can assign a number
81
70
for that particular row and column.
82
-
83
71
"""
84
-
85
72
for i in range (9 ):
86
73
for j in range (9 ):
87
74
if grid [i ][j ] == 0 :
@@ -129,17 +116,14 @@ def print_solution(grid):
129
116
"""
130
117
A function to print the solution in the form
131
118
of a 9x9 grid
132
-
133
119
"""
134
-
135
120
for row in grid :
136
121
for cell in row :
137
122
print (cell , end = " " )
138
123
print ()
139
124
140
125
141
126
if __name__ == "__main__" :
142
-
143
127
# make a copy of grid so that you can compare with the unmodified grid
144
128
for grid in (initial_grid , no_solution ):
145
129
grid = list (map (list , grid ))
0 commit comments