{-# LANGUAGE BangPatterns #-} import Control.Monad (when) import qualified Data.Vector.Fusion.Stream.Monadic as VFSM import qualified Data.Vector.Unboxed.Mutable as VUM main :: IO () main = do n <- readLn :: IO Int dp <- VUM.unsafeNew 5454 VUM.unsafeWrite dp 0 (1 :: Int) rep 7 $ \i -> rep (6 * n) $ \j -> rep' n j $ \k -> when (j >= k) $ do dpijk <- VUM.unsafeRead dp (i * 606 + j - k) VUM.unsafeModify dp (+ dpijk) ((i + 1) * 606 + j) print =<< VUM.unsafeRead dp (4848 + 6 * n) stream :: Monad m => Int -> Int -> VFSM.Stream m Int stream !l !r = VFSM.Stream step l where step x | x <= r = return $ VFSM.Yield x (x + 1) | otherwise = return $ VFSM.Done {-# INLINE [0] step #-} {-# INLINE [1] stream #-} rep :: Monad m => Int -> (Int -> m ()) -> m () rep n = flip VFSM.mapM_ (stream 0 n) stream' :: Monad m => Int -> Int -> Int -> VFSM.Stream m Int stream' !l !r !j = VFSM.Stream step l where step x | x <= j && x <= r = return $ VFSM.Yield x (x + 1) | otherwise = return $ VFSM.Done {-# INLINE [0] step #-} {-# INLINE [1] stream' #-} rep' :: Monad m => Int -> Int -> (Int -> m ()) -> m () rep' j n = flip VFSM.mapM_ (stream' 0 n j)