multi-puzzle-solver 0.9.5__py3-none-any.whl → 0.9.6__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {multi_puzzle_solver-0.9.5.dist-info → multi_puzzle_solver-0.9.6.dist-info}/METADATA +69 -1
- {multi_puzzle_solver-0.9.5.dist-info → multi_puzzle_solver-0.9.6.dist-info}/RECORD +18 -17
- puzzle_solver/__init__.py +2 -1
- puzzle_solver/puzzles/battleships/battleships.py +1 -1
- puzzle_solver/puzzles/bridges/bridges.py +1 -3
- puzzle_solver/puzzles/chess_range/chess_melee.py +0 -1
- puzzle_solver/puzzles/chess_range/chess_range.py +0 -1
- puzzle_solver/puzzles/dominosa/dominosa.py +2 -2
- puzzle_solver/puzzles/guess/guess.py +4 -3
- puzzle_solver/puzzles/inertia/inertia.py +0 -1
- puzzle_solver/puzzles/inertia/tsp.py +3 -2
- puzzle_solver/puzzles/kakurasu/kakurasu.py +46 -0
- puzzle_solver/puzzles/signpost/signpost.py +1 -3
- puzzle_solver/puzzles/stitches/stitches.py +1 -3
- puzzle_solver/puzzles/tracks/tracks.py +1 -1
- puzzle_solver/puzzles/undead/undead.py +1 -1
- {multi_puzzle_solver-0.9.5.dist-info → multi_puzzle_solver-0.9.6.dist-info}/WHEEL +0 -0
- {multi_puzzle_solver-0.9.5.dist-info → multi_puzzle_solver-0.9.6.dist-info}/top_level.txt +0 -0
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.4
|
2
2
|
Name: multi-puzzle-solver
|
3
|
-
Version: 0.9.
|
3
|
+
Version: 0.9.6
|
4
4
|
Summary: Efficient solvers for numerous popular and esoteric logic puzzles using CP-SAT
|
5
5
|
Author: Ar-Kareem
|
6
6
|
Project-URL: Homepage, https://github.com/Ar-Kareem/puzzle_solver
|
@@ -242,6 +242,11 @@ These are all the puzzles that are implemented in this repo. <br> Click on any o
|
|
242
242
|
<img src="https://raw.githubusercontent.com/Ar-Kareem/puzzle_solver/master/images/battleships_solved.png" alt="Battleships" width="140">
|
243
243
|
</a>
|
244
244
|
</td>
|
245
|
+
<td align="center">
|
246
|
+
<a href="#kakurasu-puzzle-type-30"><b>Kakurasu</b><br><br>
|
247
|
+
<img src="https://raw.githubusercontent.com/Ar-Kareem/puzzle_solver/master/images/kakurasu_solved.png" alt="Kakurasu" width="140">
|
248
|
+
</a>
|
249
|
+
</td>
|
245
250
|
</tr>
|
246
251
|
</table>
|
247
252
|
|
@@ -286,6 +291,7 @@ These are all the puzzles that are implemented in this repo. <br> Click on any o
|
|
286
291
|
- [Aquarium (Puzzle Type #27)](#aquarium-puzzle-type-27)
|
287
292
|
- [Stitches (Puzzle Type #28)](#stitches-puzzle-type-28)
|
288
293
|
- [Battleships (Puzzle Type #29)](#battleships-puzzle-type-29)
|
294
|
+
- [Kakurasu (Puzzle Type #30)](#kakurasu-puzzle-type-30)
|
289
295
|
- [Why SAT / CP-SAT?](#why-sat--cp-sat)
|
290
296
|
- [Testing](#testing)
|
291
297
|
- [Contributing](#contributing)
|
@@ -2439,6 +2445,67 @@ Time taken: 0.12 seconds
|
|
2439
2445
|
|
2440
2446
|
---
|
2441
2447
|
|
2448
|
+
## Kakurasu (Puzzle Type #30)
|
2449
|
+
|
2450
|
+
* [**Play online**](https://www.puzzle-kakurasu.com/)
|
2451
|
+
|
2452
|
+
* [**Solver Code**][30]
|
2453
|
+
|
2454
|
+
<details>
|
2455
|
+
<summary><strong>Rules</strong></summary>
|
2456
|
+
|
2457
|
+
The goal is to make some of the cells black in such a way that:
|
2458
|
+
|
2459
|
+
1. The black cells on each row sum up to the number on the right.
|
2460
|
+
|
2461
|
+
2. The black cells on each column sum up to the number on the bottom.
|
2462
|
+
|
2463
|
+
3. If a black cell is first on its row/column its value is 1. If it is second its value is 2 etc.
|
2464
|
+
|
2465
|
+
</details>
|
2466
|
+
|
2467
|
+
**Unsolved puzzle**
|
2468
|
+
|
2469
|
+
<img src="https://raw.githubusercontent.com/Ar-Kareem/puzzle_solver/master/images/kakurasu_unsolved.png" alt="Kakurasu unsolved" width="500">
|
2470
|
+
|
2471
|
+
Code to utilize this package and solve the puzzle:
|
2472
|
+
|
2473
|
+
```python
|
2474
|
+
from puzzle_solver import kakurasu_solver as solver
|
2475
|
+
side = np.array([27, 6, 1, 12, 37, 37, 11, 4, 29, 23, 66, 55])
|
2476
|
+
bottom = np.array([22, 1, 25, 36, 10, 22, 25, 35, 32, 28, 45, 45])
|
2477
|
+
binst = solver.Board(side=side, bottom=bottom)
|
2478
|
+
solutions = binst.solve_and_print()
|
2479
|
+
```
|
2480
|
+
|
2481
|
+
|
2482
|
+
**Script Output**
|
2483
|
+
|
2484
|
+
```python
|
2485
|
+
Solution found
|
2486
|
+
[['X' 'X' ' ' 'X' ' ' ' ' ' ' 'X' ' ' ' ' ' ' 'X']
|
2487
|
+
[' ' ' ' ' ' ' ' ' ' 'X' ' ' ' ' ' ' ' ' ' ' ' ']
|
2488
|
+
['X' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ']
|
2489
|
+
[' ' ' ' ' ' ' ' 'X' ' ' 'X' ' ' ' ' ' ' ' ' ' ']
|
2490
|
+
[' ' ' ' 'X' 'X' ' ' ' ' ' ' 'X' ' ' 'X' ' ' 'X']
|
2491
|
+
['X' ' ' ' ' ' ' 'X' ' ' ' ' 'X' ' ' ' ' 'X' 'X']
|
2492
|
+
[' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 'X' ' ']
|
2493
|
+
[' ' ' ' ' ' 'X' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ']
|
2494
|
+
[' ' ' ' 'X' ' ' ' ' 'X' ' ' ' ' 'X' ' ' 'X' ' ']
|
2495
|
+
[' ' ' ' ' ' 'X' ' ' ' ' 'X' ' ' ' ' ' ' ' ' 'X']
|
2496
|
+
[' ' ' ' 'X' ' ' ' ' 'X' 'X' 'X' 'X' 'X' 'X' 'X']
|
2497
|
+
['X' ' ' ' ' 'X' ' ' ' ' ' ' 'X' 'X' 'X' 'X' 'X']]
|
2498
|
+
Solutions found: 1
|
2499
|
+
status: OPTIMAL
|
2500
|
+
Time taken: 0.00 seconds
|
2501
|
+
```
|
2502
|
+
|
2503
|
+
**Solved puzzle**
|
2504
|
+
|
2505
|
+
<img src="https://raw.githubusercontent.com/Ar-Kareem/puzzle_solver/master/images/kakurasu_solved.png" alt="Kakurasu solved" width="500">
|
2506
|
+
|
2507
|
+
---
|
2508
|
+
|
2442
2509
|
---
|
2443
2510
|
|
2444
2511
|
## Why SAT / CP-SAT?
|
@@ -2519,3 +2586,4 @@ Issues and PRs welcome!
|
|
2519
2586
|
[27]: https://github.com/Ar-Kareem/puzzle_solver/tree/master/src/puzzle_solver/puzzles/aquarium "puzzle_solver/src/puzzle_solver/puzzles/aquarium at master · Ar-Kareem/puzzle_solver · GitHub"
|
2520
2587
|
[28]: https://github.com/Ar-Kareem/puzzle_solver/tree/master/src/puzzle_solver/puzzles/stitches "puzzle_solver/src/puzzle_solver/puzzles/stitches at master · Ar-Kareem/puzzle_solver · GitHub"
|
2521
2588
|
[29]: https://github.com/Ar-Kareem/puzzle_solver/tree/master/src/puzzle_solver/puzzles/battleships "puzzle_solver/src/puzzle_solver/puzzles/battleships at master · Ar-Kareem/puzzle_solver · GitHub"
|
2589
|
+
[30]: https://github.com/Ar-Kareem/puzzle_solver/tree/master/src/puzzle_solver/puzzles/kakurasu "puzzle_solver/src/puzzle_solver/puzzles/kakurasu at master · Ar-Kareem/puzzle_solver · GitHub"
|
@@ -1,19 +1,20 @@
|
|
1
|
-
puzzle_solver/__init__.py,sha256=
|
1
|
+
puzzle_solver/__init__.py,sha256=rqNCaaPZIpdL2Y_Ib96zY8u9Qs_MJaRI3CpJz_vV4_I,2189
|
2
2
|
puzzle_solver/core/utils.py,sha256=3LlBDuie_G0uSlzibpQS2ULmEYSZmpJXh1kawj7rjkg,3396
|
3
3
|
puzzle_solver/core/utils_ortools.py,sha256=qLTIzmITqmgGZvg8XpYAZ4c-lhD5sEDQfS8ECdQ_dkM,3005
|
4
4
|
puzzle_solver/puzzles/aquarium/aquarium.py,sha256=BUfkAS2d9eG3TdMoe1cOGGeNYgKUebRvn-z9nsC9gvE,5708
|
5
|
-
puzzle_solver/puzzles/battleships/battleships.py,sha256=
|
6
|
-
puzzle_solver/puzzles/bridges/bridges.py,sha256=
|
7
|
-
puzzle_solver/puzzles/chess_range/chess_melee.py,sha256=
|
8
|
-
puzzle_solver/puzzles/chess_range/chess_range.py,sha256=
|
5
|
+
puzzle_solver/puzzles/battleships/battleships.py,sha256=J1Y-zG2TOmJePCwQnUF8uE9cXL5v8GDVEGv30bQls6k,7467
|
6
|
+
puzzle_solver/puzzles/bridges/bridges.py,sha256=15A9uV4xjoqPRo_9CTnoKeGRxS3z2aMF619T1n0dTOQ,5402
|
7
|
+
puzzle_solver/puzzles/chess_range/chess_melee.py,sha256=D-_Oi8OyxsVe1j3dIKYwRlxgeb3NWLmDWGcv-oclY0c,195
|
8
|
+
puzzle_solver/puzzles/chess_range/chess_range.py,sha256=uMQGTIwzGskHIhI-tPYjT9a3wHUBIkZ18eXjV9IpUE4,21071
|
9
9
|
puzzle_solver/puzzles/chess_range/chess_solo.py,sha256=U3v766UsZHx_dC3gxqU90VbjAXn-OlYhtrnnvJYFvrQ,401
|
10
10
|
puzzle_solver/puzzles/chess_sequence/chess_sequence.py,sha256=6ap3Wouf2PxHV4P56B9ol1QT98Ym6VHaxorQZWl6LnY,13692
|
11
|
-
puzzle_solver/puzzles/dominosa/dominosa.py,sha256=
|
11
|
+
puzzle_solver/puzzles/dominosa/dominosa.py,sha256=Nmb7pn8U27QJwGy9F3wo8ylqo2_U51OAo3GN2soaNpc,7195
|
12
12
|
puzzle_solver/puzzles/filling/filling.py,sha256=UAkNYjlfxOrYGrRVmuElhWPeW10xD6kiWuB8oELzy3w,9141
|
13
|
-
puzzle_solver/puzzles/guess/guess.py,sha256=
|
14
|
-
puzzle_solver/puzzles/inertia/inertia.py,sha256=
|
15
|
-
puzzle_solver/puzzles/inertia/tsp.py,sha256=
|
13
|
+
puzzle_solver/puzzles/guess/guess.py,sha256=sH-NlYhxM3DNbhk4eGde09kgM0KaDvSbLrpHQiwcFGo,10791
|
14
|
+
puzzle_solver/puzzles/inertia/inertia.py,sha256=gJBahkh69CrSWNscalKEoP1j4X-Q3XpbIBMiG9PUpU0,5657
|
15
|
+
puzzle_solver/puzzles/inertia/tsp.py,sha256=6ZDxdg2f7GDsZeQpnSR2ay0-T4ZhTESJtkKMcm49OS0,15204
|
16
16
|
puzzle_solver/puzzles/inertia/parse_map/parse_map.py,sha256=A9JQTNqamUdzlwqks0XQp3Hge3mzyTIVK6YtDJvqpL4,8422
|
17
|
+
puzzle_solver/puzzles/kakurasu/kakurasu.py,sha256=VNGMJnBHDi6WkghLObRLhUvkmrPaGphTTUDMC0TkQvQ,2064
|
17
18
|
puzzle_solver/puzzles/keen/keen.py,sha256=tDb6C5S3Q0JAKPsdw-84WQ6PxRADELZHr_BK8FDH-NA,5039
|
18
19
|
puzzle_solver/puzzles/light_up/light_up.py,sha256=iSA1rjZMFsnI0V0Nxivxox4qZkB7PvUrROSHXcoUXds,4541
|
19
20
|
puzzle_solver/puzzles/magnets/magnets.py,sha256=-Wl49JD_PKeq735zQVMQ3XSQX6gdHiY-7PKw-Sh16jw,6474
|
@@ -23,19 +24,19 @@ puzzle_solver/puzzles/mosaic/mosaic.py,sha256=QX_nVpVKQg8OfaUcqFk9tKqsDyVqvZc6-X
|
|
23
24
|
puzzle_solver/puzzles/nonograms/nonograms.py,sha256=1jmDTOCnmivmBlwtMDyyk3TVqH5IjapzLn7zLQ4qubk,6056
|
24
25
|
puzzle_solver/puzzles/pearl/pearl.py,sha256=AP0whWwwZ-1zKingW14OwseYylNAr6NkXSrvdnPU6Rw,8566
|
25
26
|
puzzle_solver/puzzles/range/range.py,sha256=g6ZuHuulYLpNFsqbnPoIB5KoGPllYppU10-Zzqfj5f8,6993
|
26
|
-
puzzle_solver/puzzles/signpost/signpost.py,sha256
|
27
|
+
puzzle_solver/puzzles/signpost/signpost.py,sha256=-0_S6ycwzwlUf9-ZhP127Rgo5gMBOHiTM6t08dLLDac,3869
|
27
28
|
puzzle_solver/puzzles/singles/singles.py,sha256=kwMENfqQ-OP3YIz5baY6LRcvYCsNfhImEXN00lwazKM,5658
|
28
|
-
puzzle_solver/puzzles/stitches/stitches.py,sha256=
|
29
|
+
puzzle_solver/puzzles/stitches/stitches.py,sha256=iK8t02q43gH3FPbuIDn4dK0sbaOgZOnw8yHNRNvNuIU,6534
|
29
30
|
puzzle_solver/puzzles/stitches/parse_map/parse_map.py,sha256=pUB9tG__u4Fq_RCFG_4NNq3LRKAa6FRrvuuL64cUIRM,8866
|
30
31
|
puzzle_solver/puzzles/sudoku/sudoku.py,sha256=M_pry7XyKKzlfCF5rFi02lyOrj5GWZzXnDAxmD3NXvI,3588
|
31
32
|
puzzle_solver/puzzles/tents/tents.py,sha256=iyVK2WXfIT5j_9qqlQg0WmwvixwXlZSsHGK3XA-KpII,6283
|
32
33
|
puzzle_solver/puzzles/thermometers/thermometers.py,sha256=nsvJZkm7G8FALT27bpaB0lv5E_AWawqmvapQI8QcYXw,4015
|
33
34
|
puzzle_solver/puzzles/towers/towers.py,sha256=QvL0Pp-Z2ewCeq9ZkNrh8MShKOh-Y52sFBSudve68wk,6496
|
34
|
-
puzzle_solver/puzzles/tracks/tracks.py,sha256=
|
35
|
-
puzzle_solver/puzzles/undead/undead.py,sha256=
|
35
|
+
puzzle_solver/puzzles/tracks/tracks.py,sha256=0K1YZMHiRIMmFwoD_JxB2c_xB6GYV8spgNUCL-JwDJM,9073
|
36
|
+
puzzle_solver/puzzles/undead/undead.py,sha256=IrCUfzQFBem658P5KKqldG7vd2TugTHehcwseCarerM,6604
|
36
37
|
puzzle_solver/puzzles/unruly/unruly.py,sha256=sDF0oKT50G-NshyW2DYrvAgD9q9Ku9ANUyNhGSAu7cQ,3827
|
37
38
|
puzzle_solver/utils/visualizer.py,sha256=2LmNoxEqb9PGWmBmHW6jh6OqFgNYguavhHB-9Dv5EUw,6113
|
38
|
-
multi_puzzle_solver-0.9.
|
39
|
-
multi_puzzle_solver-0.9.
|
40
|
-
multi_puzzle_solver-0.9.
|
41
|
-
multi_puzzle_solver-0.9.
|
39
|
+
multi_puzzle_solver-0.9.6.dist-info/METADATA,sha256=tyAlmP1wgR8sOPLM89MHtUBv_99kGGU7_McS9m9mfRY,124310
|
40
|
+
multi_puzzle_solver-0.9.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
41
|
+
multi_puzzle_solver-0.9.6.dist-info/top_level.txt,sha256=exwVUQa-anK9vYrpKzBPvH8bX43iElWI4VeNiAyBGJY,14
|
42
|
+
multi_puzzle_solver-0.9.6.dist-info/RECORD,,
|
puzzle_solver/__init__.py
CHANGED
@@ -8,6 +8,7 @@ from puzzle_solver.puzzles.dominosa import dominosa as dominosa_solver
|
|
8
8
|
from puzzle_solver.puzzles.filling import filling as filling_solver
|
9
9
|
from puzzle_solver.puzzles.guess import guess as guess_solver
|
10
10
|
from puzzle_solver.puzzles.inertia import inertia as inertia_solver
|
11
|
+
from puzzle_solver.puzzles.kakurasu import kakurasu as kakurasu_solver
|
11
12
|
from puzzle_solver.puzzles.keen import keen as keen_solver
|
12
13
|
from puzzle_solver.puzzles.light_up import light_up as light_up_solver
|
13
14
|
from puzzle_solver.puzzles.magnets import magnets as magnets_solver
|
@@ -30,4 +31,4 @@ from puzzle_solver.puzzles.unruly import unruly as unruly_solver
|
|
30
31
|
|
31
32
|
from puzzle_solver.puzzles.inertia.parse_map.parse_map import main as inertia_image_parser
|
32
33
|
|
33
|
-
__version__ = '0.9.
|
34
|
+
__version__ = '0.9.6'
|
@@ -4,7 +4,7 @@ from typing import Optional
|
|
4
4
|
import numpy as np
|
5
5
|
from ortools.sat.python import cp_model
|
6
6
|
|
7
|
-
from puzzle_solver.core.utils import Pos, get_all_pos, get_char, get_neighbors8, set_char,
|
7
|
+
from puzzle_solver.core.utils import Pos, get_all_pos, get_char, get_neighbors8, set_char, get_row_pos, get_col_pos, get_pos, in_bounds
|
8
8
|
from puzzle_solver.core.utils_ortools import generic_solve_all, SingleSolution, or_constraint
|
9
9
|
|
10
10
|
@dataclass
|
@@ -1,12 +1,10 @@
|
|
1
|
-
import json
|
2
1
|
from collections import defaultdict
|
3
|
-
from dataclasses import dataclass
|
4
2
|
|
5
3
|
import numpy as np
|
6
4
|
from ortools.sat.python import cp_model
|
7
5
|
from ortools.sat.python.cp_model import LinearExpr as lxp
|
8
6
|
|
9
|
-
from puzzle_solver.core.utils import Pos, get_all_pos, get_char, set_char,
|
7
|
+
from puzzle_solver.core.utils import Pos, get_all_pos, get_char, set_char, get_row_pos, get_col_pos
|
10
8
|
from puzzle_solver.core.utils_ortools import generic_solve_all, SingleSolution
|
11
9
|
|
12
10
|
|
@@ -3,8 +3,8 @@ from collections import defaultdict
|
|
3
3
|
import numpy as np
|
4
4
|
from ortools.sat.python import cp_model
|
5
5
|
|
6
|
-
from puzzle_solver.core.utils import Pos,
|
7
|
-
from puzzle_solver.core.utils_ortools import
|
6
|
+
from puzzle_solver.core.utils import Pos, get_all_pos, get_char, set_char, Direction, get_next_pos, in_bounds
|
7
|
+
from puzzle_solver.core.utils_ortools import generic_solve_all, SingleSolution, or_constraint
|
8
8
|
|
9
9
|
|
10
10
|
def get_bool_var(model: cp_model.CpModel, var: cp_model.IntVar, eq_val: int, name: str) -> cp_model.IntVar:
|
@@ -1,8 +1,9 @@
|
|
1
|
-
from collections import defaultdict
|
2
|
-
import numpy as np
|
3
|
-
from collections import Counter
|
1
|
+
from collections import Counter, defaultdict
|
4
2
|
from itertools import product
|
5
3
|
|
4
|
+
import numpy as np
|
5
|
+
|
6
|
+
|
6
7
|
class Board:
|
7
8
|
def __init__(self, num_pegs: int = 4, all_colors: list[str] = ['R', 'Y', 'G', 'B', 'O', 'P'], show_warnings: bool = True, show_progress: bool = False):
|
8
9
|
assert num_pegs >= 1, 'num_pegs must be at least 1'
|
@@ -1,8 +1,10 @@
|
|
1
|
-
from collections import
|
1
|
+
from collections import defaultdict, deque
|
2
2
|
from typing import Dict, List, Tuple, Set, Any, Optional
|
3
3
|
import random
|
4
|
+
|
4
5
|
from ortools.constraint_solver import pywrapcp, routing_enums_pb2
|
5
6
|
|
7
|
+
|
6
8
|
Pos = Any # Hashable node id
|
7
9
|
|
8
10
|
def solve_optimal_walk(
|
@@ -81,7 +83,6 @@ def solve_optimal_walk(
|
|
81
83
|
prev = {n: None for n in nodes}
|
82
84
|
if src not in adj:
|
83
85
|
return dist, prev
|
84
|
-
from collections import deque
|
85
86
|
q = deque([src])
|
86
87
|
dist[src] = 0
|
87
88
|
while q:
|
@@ -0,0 +1,46 @@
|
|
1
|
+
import numpy as np
|
2
|
+
from ortools.sat.python import cp_model
|
3
|
+
|
4
|
+
from puzzle_solver.core.utils import Pos, get_all_pos, set_char, get_pos
|
5
|
+
from puzzle_solver.core.utils_ortools import generic_solve_all, SingleSolution
|
6
|
+
|
7
|
+
|
8
|
+
class Board:
|
9
|
+
def __init__(self, side: np.array, bottom: np.array):
|
10
|
+
assert side.ndim == 1, f'side must be 1d, got {side.ndim}'
|
11
|
+
self.V = side.shape[0]
|
12
|
+
assert bottom.ndim == 1, f'bottom must be 1d, got {bottom.ndim}'
|
13
|
+
self.H = bottom.shape[0]
|
14
|
+
self.side = side
|
15
|
+
self.bottom = bottom
|
16
|
+
|
17
|
+
self.model = cp_model.CpModel()
|
18
|
+
self.model_vars: dict[Pos, cp_model.IntVar] = {}
|
19
|
+
|
20
|
+
self.create_vars()
|
21
|
+
self.add_all_constraints()
|
22
|
+
|
23
|
+
def create_vars(self):
|
24
|
+
for pos in get_all_pos(self.V, self.H):
|
25
|
+
self.model_vars[pos] = self.model.NewBoolVar(f'{pos}')
|
26
|
+
|
27
|
+
def add_all_constraints(self):
|
28
|
+
for row in range(self.V):
|
29
|
+
self.model.Add(sum([self.model_vars[get_pos(x=col, y=row)] * (col + 1) for col in range(self.H)]) == self.side[row])
|
30
|
+
for col in range(self.H):
|
31
|
+
self.model.Add(sum([self.model_vars[get_pos(x=col, y=row)] * (row + 1) for row in range(self.V)]) == self.bottom[col])
|
32
|
+
|
33
|
+
def solve_and_print(self, verbose: bool = True):
|
34
|
+
def board_to_solution(board: Board, solver: cp_model.CpSolverSolutionCallback) -> SingleSolution:
|
35
|
+
assignment: dict[Pos, int] = {}
|
36
|
+
for pos, var in board.model_vars.items():
|
37
|
+
assignment[pos] = solver.value(var)
|
38
|
+
return SingleSolution(assignment=assignment)
|
39
|
+
def callback(single_res: SingleSolution):
|
40
|
+
print("Solution found")
|
41
|
+
res = np.full((self.V, self.H), ' ', dtype=object)
|
42
|
+
for pos in get_all_pos(self.V, self.H):
|
43
|
+
c = 'X' if single_res.assignment[pos] else ' '
|
44
|
+
set_char(res, pos, c)
|
45
|
+
print(res)
|
46
|
+
return generic_solve_all(self, board_to_solution, callback=callback if verbose else None, verbose=verbose)
|
@@ -1,10 +1,8 @@
|
|
1
|
-
from enum import Enum
|
2
|
-
|
3
1
|
import numpy as np
|
4
2
|
from ortools.sat.python import cp_model
|
5
3
|
from ortools.sat.python.cp_model import LinearExpr as lxp
|
6
4
|
|
7
|
-
from puzzle_solver.core.utils import Pos, get_all_pos, get_char, set_char, in_bounds, get_next_pos,
|
5
|
+
from puzzle_solver.core.utils import Pos, get_all_pos, get_char, set_char, in_bounds, get_next_pos, Direction8
|
8
6
|
from puzzle_solver.core.utils_ortools import generic_solve_all, SingleSolution
|
9
7
|
|
10
8
|
|
@@ -1,11 +1,9 @@
|
|
1
|
-
from collections import defaultdict
|
2
1
|
from typing import Union
|
3
2
|
|
4
3
|
import numpy as np
|
5
4
|
from ortools.sat.python import cp_model
|
6
|
-
from ortools.sat.python.cp_model import LinearExpr as lxp
|
7
5
|
|
8
|
-
from puzzle_solver.core.utils import Pos, get_all_pos, get_char,
|
6
|
+
from puzzle_solver.core.utils import Pos, get_all_pos, get_char, set_char, get_next_pos, Direction, get_row_pos, get_col_pos, in_bounds, get_opposite_direction
|
9
7
|
from puzzle_solver.core.utils_ortools import generic_solve_all, SingleSolution, and_constraint
|
10
8
|
|
11
9
|
|
@@ -2,7 +2,7 @@ from collections import defaultdict
|
|
2
2
|
import numpy as np
|
3
3
|
from ortools.sat.python import cp_model
|
4
4
|
|
5
|
-
from puzzle_solver.core.utils import Pos, get_all_pos, set_char, get_char,
|
5
|
+
from puzzle_solver.core.utils import Pos, get_all_pos, set_char, get_char, Direction, in_bounds, get_next_pos, get_row_pos, get_col_pos, get_opposite_direction
|
6
6
|
from puzzle_solver.core.utils_ortools import generic_solve_all, SingleSolution, and_constraint, or_constraint
|
7
7
|
|
8
8
|
|
@@ -6,7 +6,7 @@ import numpy as np
|
|
6
6
|
from ortools.sat.python import cp_model
|
7
7
|
from ortools.sat.python.cp_model import LinearExpr as lxp
|
8
8
|
|
9
|
-
from puzzle_solver.core.utils import Pos, get_all_pos, set_char, get_pos, get_next_pos, in_bounds, get_char, Direction
|
9
|
+
from puzzle_solver.core.utils import Pos, get_all_pos, set_char, get_pos, get_next_pos, in_bounds, get_char, Direction
|
10
10
|
from puzzle_solver.core.utils_ortools import generic_solve_all, SingleSolution
|
11
11
|
|
12
12
|
|
File without changes
|
File without changes
|