summaryrefslogtreecommitdiffstats
path: root/puzzle_generator.py
blob: 8159d2e2308930f2e0654bfb7049fd6990845367 (plain)
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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
from random import choices, sample
from queue import Queue
from string import ascii_lowercase
import json
from tree import Tree, get_letters
import pickle

length, height = 20, 20
word_count = 30
with open("pruned_words.json", 'r') as f:
    word_list = json.load(f)
alphabet = ascii_lowercase
vowels = "aeiouy"
weights = [8.464812190575197, 1.8290213382006486, 4.377463842181293, 3.238497204031964, 10.77374340292079,
           1.1220794892410257, 2.3639146242995768, 2.64283880414943, 8.95783107183879, 0.15584834882323761,
           0.7667338050095298, 5.577710795267362, 3.009862381185312, 7.195528282969081, 7.200021981915867,
           3.2521214123037474, 0.16781246449047607, 7.0440305216612025, 7.162211941733039, 6.607483411238427,
           3.7628002442053465, 0.945823450296398, 0.6405381447778209, 0.29978982655180736, 2.0193881642839093,
           0.4220928558487209]
root = pickle.load(open("tree.p", "rb"))

col_search_pattern = [
    [(0, 0), (0, 1), (0, 2), (0, 3), (0, 4), (0, 5), (0, 6), (0, 7), (0, 8), (0, 9), (0, 10), (0, 11), (0, 12), (0, 13),
     (0, 14), (0, 15), (0, 16), (0, 17), (0, 18), (0, 19)],
    [(1, 0), (1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (1, 6), (1, 7), (1, 8), (1, 9), (1, 10), (1, 11), (1, 12), (1, 13),
     (1, 14), (1, 15), (1, 16), (1, 17), (1, 18), (1, 19)],
    [(2, 0), (2, 1), (2, 2), (2, 3), (2, 4), (2, 5), (2, 6), (2, 7), (2, 8), (2, 9), (2, 10), (2, 11), (2, 12), (2, 13),
     (2, 14), (2, 15), (2, 16), (2, 17), (2, 18), (2, 19)],
    [(3, 0), (3, 1), (3, 2), (3, 3), (3, 4), (3, 5), (3, 6), (3, 7), (3, 8), (3, 9), (3, 10), (3, 11), (3, 12), (3, 13),
     (3, 14), (3, 15), (3, 16), (3, 17), (3, 18), (3, 19)],
    [(4, 0), (4, 1), (4, 2), (4, 3), (4, 4), (4, 5), (4, 6), (4, 7), (4, 8), (4, 9), (4, 10), (4, 11), (4, 12), (4, 13),
     (4, 14), (4, 15), (4, 16), (4, 17), (4, 18), (4, 19)],
    [(5, 0), (5, 1), (5, 2), (5, 3), (5, 4), (5, 5), (5, 6), (5, 7), (5, 8), (5, 9), (5, 10), (5, 11), (5, 12), (5, 13),
     (5, 14), (5, 15), (5, 16), (5, 17), (5, 18), (5, 19)],
    [(6, 0), (6, 1), (6, 2), (6, 3), (6, 4), (6, 5), (6, 6), (6, 7), (6, 8), (6, 9), (6, 10), (6, 11), (6, 12), (6, 13),
     (6, 14), (6, 15), (6, 16), (6, 17), (6, 18), (6, 19)],
    [(7, 0), (7, 1), (7, 2), (7, 3), (7, 4), (7, 5), (7, 6), (7, 7), (7, 8), (7, 9), (7, 10), (7, 11), (7, 12), (7, 13),
     (7, 14), (7, 15), (7, 16), (7, 17), (7, 18), (7, 19)],
    [(8, 0), (8, 1), (8, 2), (8, 3), (8, 4), (8, 5), (8, 6), (8, 7), (8, 8), (8, 9), (8, 10), (8, 11), (8, 12), (8, 13),
     (8, 14), (8, 15), (8, 16), (8, 17), (8, 18), (8, 19)],
    [(9, 0), (9, 1), (9, 2), (9, 3), (9, 4), (9, 5), (9, 6), (9, 7), (9, 8), (9, 9), (9, 10), (9, 11), (9, 12), (9, 13),
     (9, 14), (9, 15), (9, 16), (9, 17), (9, 18), (9, 19)],
    [(10, 0), (10, 1), (10, 2), (10, 3), (10, 4), (10, 5), (10, 6), (10, 7), (10, 8), (10, 9), (10, 10), (10, 11),
     (10, 12), (10, 13), (10, 14), (10, 15), (10, 16), (10, 17), (10, 18), (10, 19)],
    [(11, 0), (11, 1), (11, 2), (11, 3), (11, 4), (11, 5), (11, 6), (11, 7), (11, 8), (11, 9), (11, 10), (11, 11),
     (11, 12), (11, 13), (11, 14), (11, 15), (11, 16), (11, 17), (11, 18), (11, 19)],
    [(12, 0), (12, 1), (12, 2), (12, 3), (12, 4), (12, 5), (12, 6), (12, 7), (12, 8), (12, 9), (12, 10), (12, 11),
     (12, 12), (12, 13), (12, 14), (12, 15), (12, 16), (12, 17), (12, 18), (12, 19)],
    [(13, 0), (13, 1), (13, 2), (13, 3), (13, 4), (13, 5), (13, 6), (13, 7), (13, 8), (13, 9), (13, 10), (13, 11),
     (13, 12), (13, 13), (13, 14), (13, 15), (13, 16), (13, 17), (13, 18), (13, 19)],
    [(14, 0), (14, 1), (14, 2), (14, 3), (14, 4), (14, 5), (14, 6), (14, 7), (14, 8), (14, 9), (14, 10), (14, 11),
     (14, 12), (14, 13), (14, 14), (14, 15), (14, 16), (14, 17), (14, 18), (14, 19)],
    [(15, 0), (15, 1), (15, 2), (15, 3), (15, 4), (15, 5), (15, 6), (15, 7), (15, 8), (15, 9), (15, 10), (15, 11),
     (15, 12), (15, 13), (15, 14), (15, 15), (15, 16), (15, 17), (15, 18), (15, 19)],
    [(16, 0), (16, 1), (16, 2), (16, 3), (16, 4), (16, 5), (16, 6), (16, 7), (16, 8), (16, 9), (16, 10), (16, 11),
     (16, 12), (16, 13), (16, 14), (16, 15), (16, 16), (16, 17), (16, 18), (16, 19)],
    [(17, 0), (17, 1), (17, 2), (17, 3), (17, 4), (17, 5), (17, 6), (17, 7), (17, 8), (17, 9), (17, 10), (17, 11),
     (17, 12), (17, 13), (17, 14), (17, 15), (17, 16), (17, 17), (17, 18), (17, 19)],
    [(18, 0), (18, 1), (18, 2), (18, 3), (18, 4), (18, 5), (18, 6), (18, 7), (18, 8), (18, 9), (18, 10), (18, 11),
     (18, 12), (18, 13), (18, 14), (18, 15), (18, 16), (18, 17), (18, 18), (18, 19)],
    [(19, 0), (19, 1), (19, 2), (19, 3), (19, 4), (19, 5), (19, 6), (19, 7), (19, 8), (19, 9), (19, 10), (19, 11),
     (19, 12), (19, 13), (19, 14), (19, 15), (19, 16), (19, 17), (19, 18), (19, 19)]]
row_search_pattern = [
    [(0, 0), (1, 0), (2, 0), (3, 0), (4, 0), (5, 0), (6, 0), (7, 0), (8, 0), (9, 0), (10, 0), (11, 0), (12, 0), (13, 0),
     (14, 0), (15, 0), (16, 0), (17, 0), (18, 0), (19, 0)],
    [(0, 1), (1, 1), (2, 1), (3, 1), (4, 1), (5, 1), (6, 1), (7, 1), (8, 1), (9, 1), (10, 1), (11, 1), (12, 1), (13, 1),
     (14, 1), (15, 1), (16, 1), (17, 1), (18, 1), (19, 1)],
    [(0, 2), (1, 2), (2, 2), (3, 2), (4, 2), (5, 2), (6, 2), (7, 2), (8, 2), (9, 2), (10, 2), (11, 2), (12, 2), (13, 2),
     (14, 2), (15, 2), (16, 2), (17, 2), (18, 2), (19, 2)],
    [(0, 3), (1, 3), (2, 3), (3, 3), (4, 3), (5, 3), (6, 3), (7, 3), (8, 3), (9, 3), (10, 3), (11, 3), (12, 3), (13, 3),
     (14, 3), (15, 3), (16, 3), (17, 3), (18, 3), (19, 3)],
    [(0, 4), (1, 4), (2, 4), (3, 4), (4, 4), (5, 4), (6, 4), (7, 4), (8, 4), (9, 4), (10, 4), (11, 4), (12, 4), (13, 4),
     (14, 4), (15, 4), (16, 4), (17, 4), (18, 4), (19, 4)],
    [(0, 5), (1, 5), (2, 5), (3, 5), (4, 5), (5, 5), (6, 5), (7, 5), (8, 5), (9, 5), (10, 5), (11, 5), (12, 5), (13, 5),
     (14, 5), (15, 5), (16, 5), (17, 5), (18, 5), (19, 5)],
    [(0, 6), (1, 6), (2, 6), (3, 6), (4, 6), (5, 6), (6, 6), (7, 6), (8, 6), (9, 6), (10, 6), (11, 6), (12, 6), (13, 6),
     (14, 6), (15, 6), (16, 6), (17, 6), (18, 6), (19, 6)],
    [(0, 7), (1, 7), (2, 7), (3, 7), (4, 7), (5, 7), (6, 7), (7, 7), (8, 7), (9, 7), (10, 7), (11, 7), (12, 7), (13, 7),
     (14, 7), (15, 7), (16, 7), (17, 7), (18, 7), (19, 7)],
    [(0, 8), (1, 8), (2, 8), (3, 8), (4, 8), (5, 8), (6, 8), (7, 8), (8, 8), (9, 8), (10, 8), (11, 8), (12, 8), (13, 8),
     (14, 8), (15, 8), (16, 8), (17, 8), (18, 8), (19, 8)],
    [(0, 9), (1, 9), (2, 9), (3, 9), (4, 9), (5, 9), (6, 9), (7, 9), (8, 9), (9, 9), (10, 9), (11, 9), (12, 9), (13, 9),
     (14, 9), (15, 9), (16, 9), (17, 9), (18, 9), (19, 9)],
    [(0, 10), (1, 10), (2, 10), (3, 10), (4, 10), (5, 10), (6, 10), (7, 10), (8, 10), (9, 10), (10, 10), (11, 10),
     (12, 10), (13, 10), (14, 10), (15, 10), (16, 10), (17, 10), (18, 10), (19, 10)],
    [(0, 11), (1, 11), (2, 11), (3, 11), (4, 11), (5, 11), (6, 11), (7, 11), (8, 11), (9, 11), (10, 11), (11, 11),
     (12, 11), (13, 11), (14, 11), (15, 11), (16, 11), (17, 11), (18, 11), (19, 11)],
    [(0, 12), (1, 12), (2, 12), (3, 12), (4, 12), (5, 12), (6, 12), (7, 12), (8, 12), (9, 12), (10, 12), (11, 12),
     (12, 12), (13, 12), (14, 12), (15, 12), (16, 12), (17, 12), (18, 12), (19, 12)],
    [(0, 13), (1, 13), (2, 13), (3, 13), (4, 13), (5, 13), (6, 13), (7, 13), (8, 13), (9, 13), (10, 13), (11, 13),
     (12, 13), (13, 13), (14, 13), (15, 13), (16, 13), (17, 13), (18, 13), (19, 13)],
    [(0, 14), (1, 14), (2, 14), (3, 14), (4, 14), (5, 14), (6, 14), (7, 14), (8, 14), (9, 14), (10, 14), (11, 14),
     (12, 14), (13, 14), (14, 14), (15, 14), (16, 14), (17, 14), (18, 14), (19, 14)],
    [(0, 15), (1, 15), (2, 15), (3, 15), (4, 15), (5, 15), (6, 15), (7, 15), (8, 15), (9, 15), (10, 15), (11, 15),
     (12, 15), (13, 15), (14, 15), (15, 15), (16, 15), (17, 15), (18, 15), (19, 15)],
    [(0, 16), (1, 16), (2, 16), (3, 16), (4, 16), (5, 16), (6, 16), (7, 16), (8, 16), (9, 16), (10, 16), (11, 16),
     (12, 16), (13, 16), (14, 16), (15, 16), (16, 16), (17, 16), (18, 16), (19, 16)],
    [(0, 17), (1, 17), (2, 17), (3, 17), (4, 17), (5, 17), (6, 17), (7, 17), (8, 17), (9, 17), (10, 17), (11, 17),
     (12, 17), (13, 17), (14, 17), (15, 17), (16, 17), (17, 17), (18, 17), (19, 17)],
    [(0, 18), (1, 18), (2, 18), (3, 18), (4, 18), (5, 18), (6, 18), (7, 18), (8, 18), (9, 18), (10, 18), (11, 18),
     (12, 18), (13, 18), (14, 18), (15, 18), (16, 18), (17, 18), (18, 18), (19, 18)],
    [(0, 19), (1, 19), (2, 19), (3, 19), (4, 19), (5, 19), (6, 19), (7, 19), (8, 19), (9, 19), (10, 19), (11, 19),
     (12, 19), (13, 19), (14, 19), (15, 19), (16, 19), (17, 19), (18, 19), (19, 19)]]
l_diagonal_search_pattern = [[(19, 0)], [(18, 0), (19, 1)], [(17, 0), (18, 1), (19, 2)],
                             [(16, 0), (17, 1), (18, 2), (19, 3)], [(15, 0), (16, 1), (17, 2), (18, 3), (19, 4)],
                             [(14, 0), (15, 1), (16, 2), (17, 3), (18, 4), (19, 5)],
                             [(13, 0), (14, 1), (15, 2), (16, 3), (17, 4), (18, 5), (19, 6)],
                             [(12, 0), (13, 1), (14, 2), (15, 3), (16, 4), (17, 5), (18, 6), (19, 7)],
                             [(11, 0), (12, 1), (13, 2), (14, 3), (15, 4), (16, 5), (17, 6), (18, 7), (19, 8)],
                             [(10, 0), (11, 1), (12, 2), (13, 3), (14, 4), (15, 5), (16, 6), (17, 7), (18, 8), (19, 9)],
                             [(9, 0), (10, 1), (11, 2), (12, 3), (13, 4), (14, 5), (15, 6), (16, 7), (17, 8), (18, 9),
                              (19, 10)],
                             [(8, 0), (9, 1), (10, 2), (11, 3), (12, 4), (13, 5), (14, 6), (15, 7), (16, 8), (17, 9),
                              (18, 10), (19, 11)],
                             [(7, 0), (8, 1), (9, 2), (10, 3), (11, 4), (12, 5), (13, 6), (14, 7), (15, 8), (16, 9),
                              (17, 10), (18, 11), (19, 12)],
                             [(6, 0), (7, 1), (8, 2), (9, 3), (10, 4), (11, 5), (12, 6), (13, 7), (14, 8), (15, 9),
                              (16, 10), (17, 11), (18, 12), (19, 13)],
                             [(5, 0), (6, 1), (7, 2), (8, 3), (9, 4), (10, 5), (11, 6), (12, 7), (13, 8), (14, 9),
                              (15, 10), (16, 11), (17, 12), (18, 13), (19, 14)],
                             [(4, 0), (5, 1), (6, 2), (7, 3), (8, 4), (9, 5), (10, 6), (11, 7), (12, 8), (13, 9),
                              (14, 10), (15, 11), (16, 12), (17, 13), (18, 14), (19, 15)],
                             [(3, 0), (4, 1), (5, 2), (6, 3), (7, 4), (8, 5), (9, 6), (10, 7), (11, 8), (12, 9),
                              (13, 10), (14, 11), (15, 12), (16, 13), (17, 14), (18, 15), (19, 16)],
                             [(2, 0), (3, 1), (4, 2), (5, 3), (6, 4), (7, 5), (8, 6), (9, 7), (10, 8), (11, 9),
                              (12, 10), (13, 11), (14, 12), (15, 13), (16, 14), (17, 15), (18, 16), (19, 17)],
                             [(1, 0), (2, 1), (3, 2), (4, 3), (5, 4), (6, 5), (7, 6), (8, 7), (9, 8), (10, 9), (11, 10),
                              (12, 11), (13, 12), (14, 13), (15, 14), (16, 15), (17, 16), (18, 17), (19, 18)],
                             [(0, 0), (1, 1), (2, 2), (3, 3), (4, 4), (5, 5), (6, 6), (7, 7), (8, 8), (9, 9), (10, 10),
                              (11, 11), (12, 12), (13, 13), (14, 14), (15, 15), (16, 16), (17, 17), (18, 18), (19, 19)],
                             [(0, 1), (1, 2), (2, 3), (3, 4), (4, 5), (5, 6), (6, 7), (7, 8), (8, 9), (9, 10), (10, 11),
                              (11, 12), (12, 13), (13, 14), (14, 15), (15, 16), (16, 17), (17, 18), (18, 19)],
                             [(0, 2), (1, 3), (2, 4), (3, 5), (4, 6), (5, 7), (6, 8), (7, 9), (8, 10), (9, 11),
                              (10, 12), (11, 13), (12, 14), (13, 15), (14, 16), (15, 17), (16, 18), (17, 19)],
                             [(0, 3), (1, 4), (2, 5), (3, 6), (4, 7), (5, 8), (6, 9), (7, 10), (8, 11), (9, 12),
                              (10, 13), (11, 14), (12, 15), (13, 16), (14, 17), (15, 18), (16, 19)],
                             [(0, 4), (1, 5), (2, 6), (3, 7), (4, 8), (5, 9), (6, 10), (7, 11), (8, 12), (9, 13),
                              (10, 14), (11, 15), (12, 16), (13, 17), (14, 18), (15, 19)],
                             [(0, 5), (1, 6), (2, 7), (3, 8), (4, 9), (5, 10), (6, 11), (7, 12), (8, 13), (9, 14),
                              (10, 15), (11, 16), (12, 17), (13, 18), (14, 19)],
                             [(0, 6), (1, 7), (2, 8), (3, 9), (4, 10), (5, 11), (6, 12), (7, 13), (8, 14), (9, 15),
                              (10, 16), (11, 17), (12, 18), (13, 19)],
                             [(0, 7), (1, 8), (2, 9), (3, 10), (4, 11), (5, 12), (6, 13), (7, 14), (8, 15), (9, 16),
                              (10, 17), (11, 18), (12, 19)],
                             [(0, 8), (1, 9), (2, 10), (3, 11), (4, 12), (5, 13), (6, 14), (7, 15), (8, 16), (9, 17),
                              (10, 18), (11, 19)],
                             [(0, 9), (1, 10), (2, 11), (3, 12), (4, 13), (5, 14), (6, 15), (7, 16), (8, 17), (9, 18),
                              (10, 19)],
                             [(0, 10), (1, 11), (2, 12), (3, 13), (4, 14), (5, 15), (6, 16), (7, 17), (8, 18), (9, 19)],
                             [(0, 11), (1, 12), (2, 13), (3, 14), (4, 15), (5, 16), (6, 17), (7, 18), (8, 19)],
                             [(0, 12), (1, 13), (2, 14), (3, 15), (4, 16), (5, 17), (6, 18), (7, 19)],
                             [(0, 13), (1, 14), (2, 15), (3, 16), (4, 17), (5, 18), (6, 19)],
                             [(0, 14), (1, 15), (2, 16), (3, 17), (4, 18), (5, 19)],
                             [(0, 15), (1, 16), (2, 17), (3, 18), (4, 19)], [(0, 16), (1, 17), (2, 18), (3, 19)],
                             [(0, 17), (1, 18), (2, 19)], [(0, 18), (1, 19)], [(0, 19)]]
r_diagonal_search_pattern = [[(0, 0)], [(0, 1), (1, 0)], [(0, 2), (1, 1), (2, 0)], [(0, 3), (1, 2), (2, 1), (3, 0)],
                             [(0, 4), (1, 3), (2, 2), (3, 1), (4, 0)], [(0, 5), (1, 4), (2, 3), (3, 2), (4, 1), (5, 0)],
                             [(0, 6), (1, 5), (2, 4), (3, 3), (4, 2), (5, 1), (6, 0)],
                             [(0, 7), (1, 6), (2, 5), (3, 4), (4, 3), (5, 2), (6, 1), (7, 0)],
                             [(0, 8), (1, 7), (2, 6), (3, 5), (4, 4), (5, 3), (6, 2), (7, 1), (8, 0)],
                             [(0, 9), (1, 8), (2, 7), (3, 6), (4, 5), (5, 4), (6, 3), (7, 2), (8, 1), (9, 0)],
                             [(0, 10), (1, 9), (2, 8), (3, 7), (4, 6), (5, 5), (6, 4), (7, 3), (8, 2), (9, 1), (10, 0)],
                             [(0, 11), (1, 10), (2, 9), (3, 8), (4, 7), (5, 6), (6, 5), (7, 4), (8, 3), (9, 2), (10, 1),
                              (11, 0)],
                             [(0, 12), (1, 11), (2, 10), (3, 9), (4, 8), (5, 7), (6, 6), (7, 5), (8, 4), (9, 3),
                              (10, 2), (11, 1), (12, 0)],
                             [(0, 13), (1, 12), (2, 11), (3, 10), (4, 9), (5, 8), (6, 7), (7, 6), (8, 5), (9, 4),
                              (10, 3), (11, 2), (12, 1), (13, 0)],
                             [(0, 14), (1, 13), (2, 12), (3, 11), (4, 10), (5, 9), (6, 8), (7, 7), (8, 6), (9, 5),
                              (10, 4), (11, 3), (12, 2), (13, 1), (14, 0)],
                             [(0, 15), (1, 14), (2, 13), (3, 12), (4, 11), (5, 10), (6, 9), (7, 8), (8, 7), (9, 6),
                              (10, 5), (11, 4), (12, 3), (13, 2), (14, 1), (15, 0)],
                             [(0, 16), (1, 15), (2, 14), (3, 13), (4, 12), (5, 11), (6, 10), (7, 9), (8, 8), (9, 7),
                              (10, 6), (11, 5), (12, 4), (13, 3), (14, 2), (15, 1), (16, 0)],
                             [(0, 17), (1, 16), (2, 15), (3, 14), (4, 13), (5, 12), (6, 11), (7, 10), (8, 9), (9, 8),
                              (10, 7), (11, 6), (12, 5), (13, 4), (14, 3), (15, 2), (16, 1), (17, 0)],
                             [(0, 18), (1, 17), (2, 16), (3, 15), (4, 14), (5, 13), (6, 12), (7, 11), (8, 10), (9, 9),
                              (10, 8), (11, 7), (12, 6), (13, 5), (14, 4), (15, 3), (16, 2), (17, 1), (18, 0)],
                             [(0, 19), (1, 18), (2, 17), (3, 16), (4, 15), (5, 14), (6, 13), (7, 12), (8, 11), (9, 10),
                              (10, 9), (11, 8), (12, 7), (13, 6), (14, 5), (15, 4), (16, 3), (17, 2), (18, 1), (19, 0)],
                             [(1, 19), (2, 18), (3, 17), (4, 16), (5, 15), (6, 14), (7, 13), (8, 12), (9, 11), (10, 10),
                              (11, 9), (12, 8), (13, 7), (14, 6), (15, 5), (16, 4), (17, 3), (18, 2), (19, 1)],
                             [(2, 19), (3, 18), (4, 17), (5, 16), (6, 15), (7, 14), (8, 13), (9, 12), (10, 11),
                              (11, 10), (12, 9), (13, 8), (14, 7), (15, 6), (16, 5), (17, 4), (18, 3), (19, 2)],
                             [(3, 19), (4, 18), (5, 17), (6, 16), (7, 15), (8, 14), (9, 13), (10, 12), (11, 11),
                              (12, 10), (13, 9), (14, 8), (15, 7), (16, 6), (17, 5), (18, 4), (19, 3)],
                             [(4, 19), (5, 18), (6, 17), (7, 16), (8, 15), (9, 14), (10, 13), (11, 12), (12, 11),
                              (13, 10), (14, 9), (15, 8), (16, 7), (17, 6), (18, 5), (19, 4)],
                             [(5, 19), (6, 18), (7, 17), (8, 16), (9, 15), (10, 14), (11, 13), (12, 12), (13, 11),
                              (14, 10), (15, 9), (16, 8), (17, 7), (18, 6), (19, 5)],
                             [(6, 19), (7, 18), (8, 17), (9, 16), (10, 15), (11, 14), (12, 13), (13, 12), (14, 11),
                              (15, 10), (16, 9), (17, 8), (18, 7), (19, 6)],
                             [(7, 19), (8, 18), (9, 17), (10, 16), (11, 15), (12, 14), (13, 13), (14, 12), (15, 11),
                              (16, 10), (17, 9), (18, 8), (19, 7)],
                             [(8, 19), (9, 18), (10, 17), (11, 16), (12, 15), (13, 14), (14, 13), (15, 12), (16, 11),
                              (17, 10), (18, 9), (19, 8)],
                             [(9, 19), (10, 18), (11, 17), (12, 16), (13, 15), (14, 14), (15, 13), (16, 12), (17, 11),
                              (18, 10), (19, 9)],
                             [(10, 19), (11, 18), (12, 17), (13, 16), (14, 15), (15, 14), (16, 13), (17, 12), (18, 11),
                              (19, 10)],
                             [(11, 19), (12, 18), (13, 17), (14, 16), (15, 15), (16, 14), (17, 13), (18, 12), (19, 11)],
                             [(12, 19), (13, 18), (14, 17), (15, 16), (16, 15), (17, 14), (18, 13), (19, 12)],
                             [(13, 19), (14, 18), (15, 17), (16, 16), (17, 15), (18, 14), (19, 13)],
                             [(14, 19), (15, 18), (16, 17), (17, 16), (18, 15), (19, 14)],
                             [(15, 19), (16, 18), (17, 17), (18, 16), (19, 15)],
                             [(16, 19), (17, 18), (18, 17), (19, 16)], [(17, 19), (18, 18), (19, 17)],
                             [(18, 19), (19, 18)], [(19, 19)]]


class Tile:

    def __init__(self, x, y, letter):
        self.letter = letter
        self.neighbors = set()
        for i in range(-1, 2):
            for j in range(-1, 2):
                if not (i == 0 and j == 0):
                    tx, ty = x + i, y + j
                    if (0 <= tx < height) and (0 <= ty < length):
                        self.neighbors.add((tx, ty))


def check_word(w):
    b = root
    for i, l in enumerate(w):
        if l not in get_letters(b.children):
            return False
        b = b.children[get_letters(b.children).index(l)]
    return True


def search_string(s):
    output = []
    min_length = 5
    for l in range(min_length, len(s)+1):
        for i in range(0, len(s)-l+1):
            word = s[i:i+l]
            if check_word(word) and word in word_list:
                output.append(word)
    return output


def find_words(b):
    words = []

    for col in col_search_pattern:
        s = ""
        for coord in col:
            s += b[coord].letter
        words.append(search_string(s))
        words.append(search_string(s[::-1]))

    for row in row_search_pattern:
        s = ""
        for coord in row:
            s += b[coord].letter
        words.append(search_string(s))
        words.append(search_string(s[::-1]))

    for diag in l_diagonal_search_pattern:
        s = ""
        for coord in diag:
            s += b[coord].letter
        words.append(search_string(s))
        words.append(search_string(s[::-1]))

    for diag in r_diagonal_search_pattern:
        s = ""
        for coord in diag:
            s += b[coord].letter
        words.append(search_string(s))
        words.append(search_string(s[::-1]))

    return words


while True:
    letters = choices(alphabet, weights=weights,k=length * height)
    board = {(i, j): Tile(i, j, letters[j * height + i]) for i in range(height) for j in range(length)}
    puzzle_word_list = list(set([word for l in find_words(board) for word in l]))
    if len(puzzle_word_list) >= word_count:
        break

print(f"Total words: {len(puzzle_word_list)}")
puzzle_word_list = sample(puzzle_word_list, k=word_count)
print(f"Words: {puzzle_word_list}")

with open('puzzle.txt', "w") as f:
    f.write("Word Search Puzzle\n\n")
    for i in range(height):
        for j in range(length):
            f.write(board[(i, j)].letter + " ")
        f.write("\n")
    f.write("\n")
    for word in puzzle_word_list:
        f.write(word + "\n")

b = {coord: board[coord].letter for coord in board}
with open('board.txt', 'w') as f:
    f.write(b.__str__() + "\n")

with open('word_list.txt', 'w') as f:
    f.write(puzzle_word_list.__str__() + "\n")