結果

問題 No.17 2つの地点に泊まりたい
ユーザー aimyaimy
提出日時 2017-05-09 17:48:42
言語 Haskell
(9.8.2)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 735 bytes
コンパイル時間 135 ms
コンパイル使用メモリ 149,888 KB
最終ジャッジ日時 2024-04-27 02:25:26
合計ジャッジ時間 448 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.8.2/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )

Main.hs:2:1: error: [GHC-87110]
    Could not load module ‘Data.Map’.
    It is a member of the hidden package ‘containers-0.6.8’.
    Use -v to see a list of the files searched for.
  |
2 | import Data.Map ((!))
  | ^^^^^^^^^^^^^^^^^^^^^

Main.hs:3:1: error: [GHC-87110]
    Could not load module ‘Data.Map’.
    It is a member of the hidden package ‘containers-0.6.8’.
    Use -v to see a list of the files searched for.
  |
3 | import qualified Data.Map as M
  | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

ソースコード

diff # 

import Control.Monad
import Data.Map ((!))
import qualified Data.Map as M

main = do
 n <- readLn
 ss <- replicateM n readLn :: IO [Int]
 m <- readLn
 abcs <- map (map read . words) <$> replicateM m getLine
 print (stay2 n ss abcs)

stay2 n ss abcs = minimum ((fullSearch n ss) (warshallFloyd n abcs))

fullSearch n ss g = do
 s1 <- [1..n-2]
 s2 <- [1..n-2]
 guard (s1 /= s2)
 return (g!(0,s1) + g!(s1,s2) + g!(s2,n-1) + ss!!s1 + ss!!s2)

warshallFloyd n abcs = foldl wf g0 [(k,i,j) | k<-[0..n-1], i<-[0..n-1], j<-[0..n-1]]
 where
  g0 = foldr (\[a,b,c] g -> M.insert (b,a) c (M.insert (a,b) c g)) M.empty abcs
  wf g (k,i,j) = M.insertWith min (i,j) v g
   where v = maybe 100000 id ((+) <$> (M.lookup (i,k) g) <*> (M.lookup (k,j) g))
0