Skip to content

Commit e040dd4

Browse files
authored
Merge pull request #2576 from ERGO-Code/nqueens-correction
Minor correction to nqueens.py
2 parents 86de861 + dc852ec commit e040dd4

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/nqueens.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@
1313

1414
x = h.addBinaries(N, N)
1515

16-
h.addConstrs(x.sum(axis=0) == 1) # each row has exactly one queen
17-
h.addConstrs(x.sum(axis=1) == 1) # each col has exactly one queen
16+
h.addConstrs(x.sum(axis=1) == 1) # each row has exactly one queen
17+
h.addConstrs(x.sum(axis=0) == 1) # each col has exactly one queen
1818

1919
y = np.fliplr(x)
2020
h.addConstrs(x.diagonal(k).sum() <= 1 for k in range(-N + 1, N)) # each diagonal has at most one queen
@@ -26,4 +26,4 @@
2626
print('Queens:')
2727

2828
for i in range(N):
29-
print(''.join('Q' if sol[i, j] > 0.5 else '*' for j in range(N)))
29+
print(''.join('Q' if sol[i, j] > 0.5 else '*' for j in range(N)))

0 commit comments

Comments
 (0)