結果

問題 No.561 東京と京都
ユーザー atuk721atuk721
提出日時 2017-11-13 06:28:24
言語 Haskell
(9.8.2)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 1,249 bytes
コンパイル時間 5,758 ms
コンパイル使用メモリ 175,176 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-05-03 20:04:18
合計ジャッジ時間 4,210 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 AC 1 ms
5,376 KB
testcase_07 AC 1 ms
5,376 KB
testcase_08 AC 1 ms
5,376 KB
testcase_09 AC 1 ms
5,376 KB
testcase_10 AC 1 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 3 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 3 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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