[Haskell-beginners] Partition a number or optimize