結果

問題 No.561 東京と京都
ユーザー Asdf_QwertyZAsdf_QwertyZ
提出日時 2017-09-03 00:19:55
言語 Haskell
(9.10.1)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 788 bytes
コンパイル時間 5,522 ms
コンパイル使用メモリ 170,880 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-06 18:38:33
合計ジャッジ時間 6,396 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 # 

import Control.Applicative ((<$>))
import Control.Monad (replicateM)

data Pos = Tokyo | Kyoto
  deriving Eq

flipPos :: Pos -> Pos
flipPos Tokyo = Kyoto
flipPos Kyoto = Tokyo

toIdx :: Pos -> Int
toIdx Tokyo = 0
toIdx Kyoto = 1

getNums :: IO [Int]
getNums = map read . words <$> getLine

{-
  dp[i][k] := i までで位置 k で何円か
  dp[i+1][k] := max( dp[i][k]+TK[k][i], dp[i][!k]+Tk[k][i]-d )
-}

main :: IO ()
main = do
  [n, d] <- getNums :: IO [Int]
  ([iniT,iniK]:tks) <- replicateM n getNums

  print $ (\[a,b] -> max a b) $ foldl (f d) [iniT,iniK-d] tks
  
f :: Int -> [Int] -> [Int] -> [Int]
f d dps tks = [dp Tokyo, dp Kyoto] where
  dp pos = (dps!!curIdx + tks!!curIdx) `max` (dps!!revIdx + tks!!curIdx - d) where
    curIdx = toIdx pos
    revIdx = toIdx $ flipPos pos
0