neingeist/python-exercises
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

n-queens: use list comprehension instead of for loop

Browse source
This commit is contained in:
neingeist 2015-05-30 21:10:28 +02:00
parent a4249017c0
commit 7ee2c4ad8f
1 changed files with 2 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

7
n-queens.py
View file

@ -32,11 +32,8 @@ def solve(n):
positions[row]-k != positions[row+k]
for row in range(0, n-1) for k in range(1, n-row)))
solutions = []
for positions in permutations(range(1, n+1)):
if valid(positions):
solutions.append(positions)
return solutions
return [positions for positions in permutations(range(1, n+1))
if valid(positions)]
def fundamental(solutions):