結果

問題 No.406 鴨等間隔の法則
コンテスト
ユーザー はむ吉🐹
提出日時 2016-07-18 20:22:08
言語 Haskell
(9.10.1)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,146 bytes
コンパイル時間 3,415 ms
コンパイル使用メモリ 148,992 KB
最終ジャッジ日時 2024-07-07 11:15:32
合計ジャッジ時間 3,859 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、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:4:1: error: [GHC-87110]
    Could not load module ‘Data.IntSet’.
    It is a member of the hidden package ‘containers-0.6.8’.
    Use -v to see a list of the files searched for.
  |
4 | import qualified Data.IntSet as S
  | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

ソースコード

diff # 

-- #101409よりも高速であるはずのHaskell解

import Control.Applicative ((<$>))
import qualified Data.IntSet as S
import qualified Data.ByteString.Lazy.Char8 as C
import Data.List (sort)
import Data.Maybe (fromJust)

-- 与えられた数列の階差数列を求める関数
-- WriterによるPython解と同様の考え方による
diff :: Num a => [a] -> [a]
diff xs = zipWith (-) (tail xs) (init xs)

-- 与えられた数列が二条件を同時に満たすかを判定する関数
isKamo :: [Int] -> Bool
isKamo = f . S.fromList . diff . sort
    where f s = S.size s == 1 && S.notMember 0 s

-- 遅延ByteStringからIntを読み取る関数
unsafeReadInt :: C.ByteString -> Int
unsafeReadInt = fst . fromJust . C.readInt

main :: IO ()
main = do
    -- 入力全体を読み取り、単語ごとに分割してそれぞれ整数に変換する
    -- 最初の要素であるNは使わないので、tailで捨てる
    -- Stringではなく遅延ByteStringで入力を受け取り、高速化を図る
    xs <- fmap unsafeReadInt . tail . C.words <$> C.getContents :: IO [Int]
    putStrLn $ if isKamo xs then "YES" else "NO"
0