In today’s Programming Praxis we have to solve the classic n-queens problem. The provided Scheme solution has 13 lines. Let’s see if we can do any better.

A quick import:

import Data.List

The wikipedia page for the algorithm mentions a simple algorithm where you take the permutations of 1 through n as the column positions for the n consecutive rows and removing the illegal ones. So let’s see if that works.

queens :: Int -> [[Int]]
queens n = filter (safe . zip [1..]) $ permutations [1..n]

Since the algorithm guarantees that no two queens will be on the same row or column, we only need to check the diagonals.

safe :: [(Int, Int)] -> Bool
safe [] = True
safe (x:xs) = all (safe' x) xs && safe xs where
safe' (x1,y1) (x2,y2) = x1+y1 /= x2+y2 && x1-y1 /= x2-y2

A quick test produces the same results as the Scheme solution, and the correct amount according to Wikipedia. At four lines, that will do nicely (you can make it 3 by expressing safe as safe xs = and . zipWith (all . safe’) xs . tail $ tails xs, but I find that version to be less clear than the current one).

main :: IO ()
main = mapM_ print $ queens 5

### Like this:

Like Loading...

*Related*

Tags: bonsai, chess, code, Haskell, kata, praxis, programming, queens

This entry was posted on June 11, 2010 at 1:27 pm and is filed under Programming Praxis. You can follow any responses to this entry through the RSS 2.0 feed.
You can leave a response, or trackback from your own site.

## Leave a Reply