結果

問題 No.561 東京と京都
ユーザー atuk721atuk721
提出日時 2017-11-13 06:28:24
言語 Haskell
(9.6.2)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 1,249 bytes
コンパイル時間 2,557 ms
コンパイル使用メモリ 159,948 KB
実行使用メモリ 8,312 KB
最終ジャッジ日時 2023-08-16 11:35:35
合計ジャッジ時間 3,972 ms
ジャッジサーバーID
(参考情報)
judge15 / judge12
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,888 KB
testcase_01 AC 2 ms
6,856 KB
testcase_02 AC 3 ms
6,968 KB
testcase_03 AC 3 ms
6,872 KB
testcase_04 AC 2 ms
6,980 KB
testcase_05 AC 2 ms
6,824 KB
testcase_06 AC 3 ms
6,868 KB
testcase_07 AC 2 ms
6,904 KB
testcase_08 AC 3 ms
6,944 KB
testcase_09 AC 3 ms
7,024 KB
testcase_10 AC 2 ms
7,020 KB
testcase_11 AC 3 ms
7,216 KB
testcase_12 AC 4 ms
8,104 KB
testcase_13 AC 4 ms
8,220 KB
testcase_14 AC 3 ms
7,660 KB
testcase_15 AC 4 ms
8,240 KB
testcase_16 AC 4 ms
8,272 KB
testcase_17 AC 4 ms
7,944 KB
testcase_18 AC 4 ms
8,284 KB
testcase_19 AC 4 ms
8,292 KB
testcase_20 AC 4 ms
8,312 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.6.1/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