Haskell Quiz/Weird Numbers/Solution Ninju

From HaskellWiki
< Haskell Quiz‎ | Weird Numbers
Revision as of 07:51, 13 December 2009 by Newacct (talk | contribs)
(diff) ← Older revision | Latest revision (diff) | Newer revision → (diff)
Jump to navigation Jump to search


module Main where
import System.Environment

main :: IO ()
main = do args <- getArgs
          let max = read (head args)
          if length args == 1
            then do putStrLn $ "The weird numbers up to " ++ (head args) ++ ":"
                    print $ takeWhile (<= max) weirdNumbers
            else putStrLn "Usage: weird_numbers LIMIT"

weirdNumbers :: [Integer]
weirdNumbers = filter weird [70..]
               where
               weird n = sum (divisors n) > n && (not . any (==n) . map sum . subsets . divisors) n

divisors :: Integer -> [Integer]
divisors n = filter ((==0) . (mod n)) [1..(n `div` 2)]

subsets :: [a] -> [[a]]
subsets [] = [[]]
subsets (x:xs) = subsets xs ++ map (x:) (subsets xs)