結果

問題 No.561 東京と京都
ユーザー atuk721
提出日時 2017-11-13 06:28:24
言語 Haskell
(9.10.1)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 1,249 bytes
コンパイル時間 12,182 ms
コンパイル使用メモリ 172,928 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-25 00:00:45
合計ジャッジ時間 4,350 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 17
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.8.2/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
[2 of 2] Linking a.out

ソースコード

diff # 

{-# LANGUAGE RecordWildCards #-}

import Control.Monad.State.Lazy
import Data.List

data Question = Question { days :: Int, moveCost :: Int, tokyoIncome :: [Int], kyotoIncome :: [Int] } deriving Show
data CalcState = CalcState { maxTokyo :: Int, maxKyoto :: Int } deriving Show

input :: IO Question
input = do
  [d, c] <- getLineData
  [tx, kx] <- fmap transpose $ sequence $ replicate d getLineData
  return Question { days = d, moveCost = c, tokyoIncome = tx, kyotoIncome = kx }
    where getLineData = fmap (map read . words) getLine

process :: Question -> Int
process Question {..} = let CalcState {..} = execState (process' 1) $ CalcState { maxTokyo = 0, maxKyoto = -moveCost }
                         in max maxTokyo maxKyoto
  where
    process' :: Int -> State CalcState ()
    process' d
      | d > days = return ()
      | otherwise = do
          modify $ \CalcState {..} ->
            let (maxTokyo', maxKyoto') =
                  (tokyoIncome !! (d - 1) + max maxTokyo (maxKyoto - moveCost),
                   kyotoIncome !! (d - 1) + max (maxTokyo - moveCost) maxKyoto)
             in CalcState { maxTokyo = maxTokyo', maxKyoto = maxKyoto' }
          process' (d + 1)

main :: IO ()
main = input >>= return . process >>= print
0