simplify nqueens benchmark

2622b6dc · Stefan Behnel · 9d77b151 · 2622b6dc
Commit 2622b6dc authored May 15, 2011 by Stefan Behnel
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

Demos/benchmarks/nqueens.py Demos/benchmarks/nqueens.py +2 -2

No files found.
--- a/Demos/benchmarks/nqueens.py
+++ b/Demos/benchmarks/nqueens.py
@@ -58,8 +58,8 @@ def n_queens(queen_count):
    """
    cols = range(queen_count)
    for vec in permutations(cols):
-        if (queen_count == len(set([vec[i]+i for i in cols]))
-                        == len(set([vec[i]-i for i in cols]))):
+        if (queen_count == len({ vec[i]+i for i in cols })
+                        == len({ vec[i]-i for i in cols })):
            yield vec