Created
December 6, 2012 02:41
-
-
Save bgschiller/4221410 to your computer and use it in GitHub Desktop.
Brute force solution to Ninja Assassin Wonderwall problem
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import Data.List (elemIndex) | |
-- | The 'permutations' function returns the list of all permutations of the argument. | |
-- | |
-- > permutations "abc" == ["abc","bac","cba","bca","cab","acb"] | |
permutations :: [a] -> [[a]] | |
permutations xs0 = xs0 : perms xs0 [] | |
where | |
perms [] _ = [] | |
perms (t:ts) is = foldr interleave (perms ts (t:is)) (permutations is) | |
where interleave xs r = let (_,zs) = interleave' id xs r in zs | |
interleave' _ [] r = (ts, r) | |
interleave' f (y:ys) r = let (us,zs) = interleave' (f . (y:)) ys r | |
in (y:us, f (t:y:us) : zs) | |
{-players is a list of the players. | |
permutations is a list of triplets, one for each player p: (p, p's wonderwall, p's ninja assassin)-} | |
validOrderings::(Show a, Ord a) => [a] -> [(a,a,a)] -> [[a]] | |
validOrderings players preferences = filter validOrder (permutations players) | |
where | |
validOrder possible = not (elem False | |
(map (\ (x, y, z) -> ((compare (elemIndex x possible) (elemIndex y possible)) == | |
(compare (elemIndex y possible) (elemIndex z possible)))) preferences)) | |
main = do | |
putStr $ (show results) ++ "\n" | |
putStr $ (show results2) ++ "\n" | |
where | |
results = validOrderings ['a','b','c','d','e','f'] [('a','b','c'),('b','c','d'),('c','b','a'),('d','b','a'),('e','a','d'),('f','c','e')] | |
results2 = validOrderings [1,2,3,4,5,6,7] [(1,2,3),(2,4,6),(3,4,7),(4,1,5),(5,6,3),(6,1,7),(7,2,5)] | |
{- Example usage: | |
validOrderings ['a','b','c','d','e','f'] [('a','b','c'),('b','c','d'),('c','b','a'),('d','b','a'),('e','a','d'),('f','c','e')] | |
outputs ["eabcdf","fdcbae","dfcbae","eabcfd"] | |
validOrderings [1,2,3,4,5,6,7] [(1,2,3),(2,4,6),(3,4,7),(4,1,5),(5,6,3),(6,1,7),(7,2,5)] -} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment