{-# LANGUAGE BangPatterns #-} import qualified Data.Vector.Fusion.Stream.Monadic as VFSM import qualified Data.Vector.Unboxed.Mutable as VUM modulo :: Int modulo = 1000000007 {-# INLINE modulo #-} main :: IO () main = do n <- readLn :: IO Int dp <- VUM.unsafeNew 3000003 :: IO (VUM.IOVector Int) VUM.unsafeWrite dp 0 1 VUM.unsafeWrite dp 1 0 VUM.unsafeWrite dp 2 0 VUM.unsafeWrite dp 3 1 VUM.unsafeWrite dp 4 0 VUM.unsafeWrite dp 5 0 VUM.unsafeWrite dp 6 0 VUM.unsafeWrite dp 7 1 VUM.unsafeWrite dp 8 0 rep n $ \i -> do item1 <- VUM.unsafeRead dp ((i - 1) * 3 + 1) item2 <- VUM.unsafeRead dp ((i - 2) * 3) item3 <- VUM.unsafeRead dp ((i - 3) * 3) item4 <- VUM.unsafeRead dp ((i - 1) * 3 + 2) item5 <- VUM.unsafeRead dp ((i - 2) * 3 + 2) item6 <- VUM.unsafeRead dp ((i - 3) * 3 + 1) VUM.unsafeWrite dp (i * 3) (item1 + item4) VUM.unsafeWrite dp ((i * 3) + 1) (item2 + item5) VUM.unsafeWrite dp ((i * 3) + 2) (item3 + item6) VUM.unsafeModify dp (`mod` modulo) (i * 3) VUM.unsafeModify dp (`mod` modulo) (i * 3 + 1) VUM.unsafeModify dp (`mod` modulo) (i * 3 + 2) item1 <- VUM.unsafeRead dp (n * 3) item2 <- VUM.unsafeRead dp (n * 3 + 1) item3 <- VUM.unsafeRead dp (n * 3 + 2) print . flip mod modulo $ item1 + item2 + item3 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 3 n) {-# INLINE rep #-}