結果

問題 No.517 壊れたアクセサリー
ユーザー aimy
提出日時 2017-05-29 15:22:13
言語 Haskell
(9.10.1)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 860 bytes
コンパイル時間 8,237 ms
コンパイル使用メモリ 176,140 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-21 18:09:45
合計ジャッジ時間 6,984 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 15
権限があれば一括ダウンロードができます
コンパイルメッセージ
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:21:9: warning: [GHC-63394] [-Wx-partial]
    In the use of ‘head’
    (imported from Data.List, but defined in GHC.List):
    "This is a partial function, it throws an error on empty lists. Use pattern matching or Data.List.uncons instead. Consider refactoring to use Data.List.NonEmpty."
   |
21 |   pre = head a
   |         ^^^^
[2 of 2] Linking a.out

ソースコード

diff #

import Control.Monad
import Data.Maybe
import Data.List

last' xs = if null xs then Nothing else Just (last xs)

main = do
 n <- readLn
 as <- replicateM n getLine
 m <- readLn
 bs <- replicateM m getLine
 putStrLn (maybe "-1" id (brokenAcc as bs))

brokenAcc as bs = do
 let as' = map (checkps bs) as
 guard (valid as')
 return (connect as')

checkps bs a = (a, (prev,next))
 where
  pre = head a
  suf = last a
  prev = find (elem pre) bs >>= last' . takeWhile (/= pre)
  next = find (elem suf) bs >>= last' . takeWhile (/= suf) . reverse

valid = (\(ps,ss) -> (nothing ps, nothing ss) == (1,1)) . unzip . map snd
 where nothing = length . filter isNothing

connect as = concat (unfoldr connect' (as,Nothing))
 where
  connect' ([],_) = Nothing
  connect' (as,next) = Just (a, (as',Just (last a)))
   where ([(a,(_,s))],as') = partition ((==next).fst.snd) as
0