結果

問題 No.17 2つの地点に泊まりたい
ユーザー aimyaimy
提出日時 2017-05-09 17:48:42
言語 Haskell
(9.6.2)
結果
AC  
実行時間 494 ms / 5,000 ms
コード長 735 bytes
コンパイル時間 5,000 ms
コンパイル使用メモリ 167,156 KB
実行使用メモリ 132,548 KB
最終ジャッジ日時 2023-08-05 03:52:21
合計ジャッジ時間 10,596 ms
ジャッジサーバーID
(参考情報)
judge13 / judge12
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
7,540 KB
testcase_01 AC 3 ms
7,456 KB
testcase_02 AC 6 ms
9,984 KB
testcase_03 AC 444 ms
101,716 KB
testcase_04 AC 32 ms
17,900 KB
testcase_05 AC 303 ms
90,644 KB
testcase_06 AC 123 ms
39,184 KB
testcase_07 AC 102 ms
37,448 KB
testcase_08 AC 454 ms
101,868 KB
testcase_09 AC 454 ms
101,820 KB
testcase_10 AC 383 ms
97,768 KB
testcase_11 AC 494 ms
132,548 KB
testcase_12 AC 3 ms
7,456 KB
testcase_13 AC 2 ms
7,556 KB
testcase_14 AC 3 ms
7,488 KB
testcase_15 AC 3 ms
7,504 KB
testcase_16 AC 3 ms
7,448 KB
testcase_17 AC 32 ms
17,688 KB
testcase_18 AC 364 ms
101,896 KB
testcase_19 AC 344 ms
102,780 KB
testcase_20 AC 22 ms
15,236 KB
testcase_21 AC 5 ms
9,432 KB
testcase_22 AC 424 ms
101,792 KB
testcase_23 AC 242 ms
65,060 KB
testcase_24 AC 305 ms
70,100 KB
testcase_25 AC 5 ms
9,480 KB
testcase_26 AC 425 ms
99,828 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 #

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